bzoj3378[Usaco2004 Open]MooFest 狂欢节*

bzoj3378[Usaco2004 Open]MooFest 狂欢节

题意：

n只奶牛，第i只听力为vi，坐标为xi，两只奶牛聊天时音量是max(vi,vj)*abs(xi-xj)。求n(n-1)/2对奶牛的音量和。n≤20000。

题解：

首先所有奶牛按x排序，记录其位置，接着再按它们音量升序排序依次插入树状数组。维护两个树状数组，一个用来求位置比某奶牛大的坐标和和奶牛数，另一个用来求位置比某奶牛小的坐标和和奶牛数。对于每个插入的奶牛i，对答案的贡献是vi*位置比它大的坐标和与奶牛数*该奶牛的坐标的差，加上vi*位置比它小的坐标和与奶牛数*该奶牛的坐标的差的相反数。

代码：

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <queue>
 #define inc(i,j,k) for(int i=j;i<=k;i++)
 #define maxn 20010
 #define ll long long
 #define lb(x) x&-x
 using namespace std;

 inline int read(){
     char ch=getchar(); int f=,x=;
     while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();}
     while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
     return f*x;
 }
 struct nd{int d,sz;}nds1[maxn],nds2[maxn]; int n; ll ans;
 struct abc{int v,x,id;}abcd[maxn];
 bool cmp1(abc a,abc b){return a.x<b.x;} bool cmp2(abc a,abc b){return a.v<b.v;}
 void update1(int x,int y){while(x<=n)nds1[x].d+=y,nds1[x].sz++,x+=lb(x);}
 void update2(int x,int y){while(x)nds2[x].d+=y,nds2[x].sz++,x-=lb(x);}
 nd query1(int x){nd q=(nd){,}; while(x>=)q.d+=nds1[x].d,q.sz+=nds1[x].sz,x-=lb(x); return q;}
 nd query2(int x){nd q=(nd){,}; while(x<=n)q.d+=nds2[x].d,q.sz+=nds2[x].sz,x+=lb(x); return q;}
 int main(){
     n=read(); inc(i,,n)abcd[i].v=read(),abcd[i].x=read(); sort(abcd+,abcd+n+,cmp1);
     inc(i,,n)abcd[i].id=i; sort(abcd+,abcd+n+,cmp2);
     inc(i,,n){
         nd a=query1(abcd[i].id-); ans+=(ll)abcd[i].v*(abcd[i].x*a.sz-a.d);
         a=query2(abcd[i].id+); ans+=(ll)abcd[i].v*(a.d-abcd[i].x*a.sz);
         update1(abcd[i].id,abcd[i].x); update2(abcd[i].id,abcd[i].x);
     }
     printf("%lld",ans); return ;
 }

20161116