POJ 1990 MooFest 题解 《挑战程序设计竞赛》

#include <iostream>#include<cstdio>#include<algorithm>using namespace std;typedef long long ll;const int MAX=20005;int n;pair<int,int> cow[MAX];ll num[MAX],dis[MAX];ll sum(ll *bit,int i){    ll s=0;    while(i>0)    {        s+=bit[i];        i-=i&(-i);    }    return s;}ll sum(ll *bit,int from,int to){    return sum(bit,to-1)-sum(bit,from-1);}void add(ll *bit,int i,ll x){    while(i<=MAX)    {        bit[i]+=x;        i+=i&(-i);    }}int main(){    while(cin>>n)    {        for(int i=0;i<n;i++)        cin>>cow[i].first>>cow[i].second;        sort(cow,cow+n);        ll res=0;        for(int i=0;i<n;i++)        {            int h=cow[i].first,x=cow[i].second;             ll left=sum(num,1,x),right=i-left;            res+=h*((left*x-sum(dis,1,x))+(sum(dis,x+1,MAX)-right*x));            add(num,x,1);            add(dis,x,x);        }        cout<<res<<endl;    }    return 0;}

好久未做线段树之类的题了，感觉生疏了许多，参考了一下别人的代码。

首先这一题使用了两个BIT，一个用来求左右两边的cow的个数，另外一个是用来记录所有点到原点的距离之和。

此外，此题先通过听力好坏程度排序，这样每次新更新的牛都是耳背的，不用再比较两个cow的的听力值了。