POJ--1990(排序+树状数组)

#include <cstdio>
#include<cstring>
#include<map>
#include<algorithm>
#include<vector>
#include<iostream>
using namespace std;
typedef long long ll;
const int maxn = 2e4+100;
ll num[maxn],ds[maxn];
int n;
struct node
{
    int v,x;
}st[maxn];
bool cmp(node a,node b){
    return a.v<b.v;
}
ll lowbit(ll x)
{
    return x&(-x);
}
ll getsum(int id,ll c[])
{
    ll an=0;
    for(int i=id;i>0;i-=lowbit(i))
        an+=c[i];
    return an;
}
void update(ll c[],int id,ll x)
{
    for(int i=id;i<maxn;i+=lowbit(i))
        c[i]+=x;
    return ;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&st[i].v,&st[i].x);
    ll all=0;
    ll tot=0;
    sort(st+1,st+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        ll sx=getsum(st[i].x,num);
        ll sm=getsum(st[i].x,ds);
        ll small=sx*st[i].x-sm;
        ll big=tot-sm-(i-1-sx)*st[i].x;
        all=all+(small+big)*st[i].v;　　//记录答案
        tot+=st[i].x;　　//更新i之前的和值
        update(ds,st[i].x,st[i].x);//更新比它x小和
        update(num,st[i].x,1);//更新比它x小的数目
    }
    cout<<all<<endl;
}