poj 1990

题目链接

借鉴cxlove大神的思路

题意：听力v，位置x，2个牛交流声音为max(v1,v2)*(x1-x2),求总的

10000^2 tle

用的树状数组做的，排序，2个，小于vi的牛的总数和距离总和

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define maxn 20005
int arr[maxn],s[maxn];
//pair<int , int> cow[MAXN]; 类似结构体
struct tree
{
    int x,v;
}a[maxn];
int cmp(tree a,tree b)
{
    return a.v<b.v;
}
int lowbit(int x)
{
    return x&-x;
}
void update(int *a,int x,int val)
{
    while(x<maxn)
    {
        a[x]+=val;
        x+=lowbit(x);
    }
}
int getsum(int *a,int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=a[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    int n,i,j;
    while(scanf("%d",&n)!=EOF)
    {
        memset(arr,0,sizeof(arr));
        memset(s,0,sizeof(s));
        for(i=0;i<n;i++) scanf("%d%d",&a[i].v,&a[i].x);
        __int64 ans=0,total=0;
        sort(a,a+n,cmp);
        for(i=0;i<n;i++Q){
            update(arr,a[i].x,1);
            update(s,a[i].x,a[i].x);
            total+=a[i].x;
            __int64 s_num=getsum(arr,a[i].x);
            __int64 s_x=getsum(s,a[i].x);
            ans+=a[i].v*(s_num*a[i].x-s_x+total-s_x-(i+1-s_num)*a[i].x);

        }
        printf("%I64d\n",ans);
    }
    return 0;
}