zoukankan      html  css  js  c++  java
  • POJ 1990 MooFest 树状数组

      这题是我看了大白书树状数组后刷的第一道题,确实难度不小,所以只好上网找题解了,网上的做法确实精彩。这题的题意主要是有N头牛,每两头牛之间交流的费用为它们的距离乘上两者音量的最大值(即max(v(i),v(j))),然后统计所有牛两两交流的总费用。一开始能想到的做法便是O(n2)的暴力枚举了,当时的我也只能想到这样的复杂度,很纳闷怎么能和树状数组搭上边呢?然后看了别人的题解后才惊叹其思路之妙。

      在博客 http://www.cnblogs.com/Fatedayt/archive/2011/10/08/2202439.html 上说得很详细,附上其主要的思路:

      已经说得很详细了,即统计某头牛i 时,通过排序和预处理可以把原本的O(n)查询下降到O(logn),确实很厉害,也很难想到。我结合网上其他人的题解半抄半改写出如下代码:

     1 #include<cstdio>
     2 #include<cstring>
     3 #include<algorithm>
     4 using namespace std;
     5 typedef long long LL;
     6 const int maxn= 20020;
     7 
     8 inline LL lowbit(LL x)    {    return x&(-x);    }
     9 
    10 struct treeArray{
    11     LL c[maxn], n;
    12     treeArray(LL n= 0): n(n) {    memset(c,0,sizeof(c));    }
    13     LL sum(LL x) {
    14         LL ans= 0;
    15         while(x){
    16             ans+= c[x];
    17             x-= lowbit(x);
    18         }
    19         return ans;
    20     }
    21     void add(LL x, LL d){
    22         while(x<=n){
    23             c[x]+= d;
    24             x+= lowbit(x);
    25         }
    26     }
    27 } Count(20003),dist(20003);
    28 
    29 struct Cow{
    30     LL v,x;
    31     bool operator <(const Cow c2) const {
    32         return v<c2.v;
    33     }
    34 } cow[maxn];
    35 
    36 int main(){
    37     int n,i;
    38     scanf("%d",&n);
    39     for(i=1; i<=n; ++i)
    40         scanf("%lld%lld",&cow[i].v,&cow[i].x);
    41     sort(cow+1,cow+n+1);
    42     LL ans= 0, alldist= 0;
    43     for(i=1; i<=n; ++i){
    44         LL x = cow[i].x;
    45         LL num = Count.sum(x);
    46         LL lessdist = dist.sum(x);
    47         ans += cow[i].v*(num*x-lessdist+alldist-lessdist-(i-1-num)*x);
    48         Count.add(x,1);
    49         dist.add(x,x);
    50         alldist += x;
    51     }
    52     printf("%lld
    ",ans);
    53     return 0;
    54 }

      其中,Count.sum(x)表示统计比x小的牛的个数(在上图中就是a),dist.sum(x)表示统计比x小的牛的位置之和(在上图中为b);alldist即前i 头牛的所有距离(可以认为是以原点作为标准),其余的变量名也不难理解。码完后才感到树状数组真是太强大了,不得不赞!

  • 相关阅读:
    docker-compose,docker-machine,docker swarm 的简单总结
    linux ubuntu安装好后,开通远程登录
    docker数据卷的使用 -v --volumes--from
    shell脚本中source无效
    模块 shutil
    模块 sys
    模块 os
    模块 random
    模块 datetime,time
    import本质
  • 原文地址:https://www.cnblogs.com/Newdawn/p/4161986.html
Copyright © 2011-2022 走看看