树状数组的修炼 疑惑篇

洛谷p2345

奶牛叫不出の痛

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 2e5 + 10;
int n;
ll cnt[maxn];//cnt[i]表示i位置及其左侧牛的个数 
ll sum[maxn];
ll res = 0;
ll mx = 0;
struct cow
{
    ll x, v;
    bool operator < (const cow &a){
        return v < a.v;
    }
};
cow a[maxn]; 

int lowbit(int x)
{
    return x & -x;
}

void updcnt(int x, int val)
{
    for(int i = x ; i <= mx ; i += lowbit(i)){
        cnt[i] += val;
    }
}

void updsum(int x, ll val)
{
    for(int i = x ; i <= mx ; i += lowbit(i)){
        sum[i] += val;
    }
}

ll getcnt(int x)
{
    ll tmp = 0;
    for(int i = x ; i >= 1 ; i -= lowbit(i)){
        tmp += cnt[i];
    }
    return tmp;
}

ll getsum(int x)//求原点到x的所有存在的点的距离 
{
    ll s = 0;
    for(int i = x ; i >= 1 ; i -= lowbit(i)){
        s += sum[i];
    }
    return s;
}

int main(){
    scanf("%d",&n);
    for(int i = 1 ; i <= n ; i++){
        scanf("%lld%lld",&a[i].v,&a[i].x);
        mx = max(mx, a[i].x);//最远距离 
    }
    sort(a + 1, a + n + 1);
    
    updcnt(a[1].x, 1);
    updsum(a[1].x, a[1].x);
    for(int i = 2 ; i <= n ; i++){//
        updcnt(a[i].x, 1);
        updsum(a[i].x, a[i].x);
        ll cl = getcnt(a[i].x - 1), cr = getcnt(mx) - getcnt(a[i].x);//cnt of left/right
        ll sl = getsum(a[i].x - 1), sr = getsum(mx) - getsum(a[i].x);//sum of left/right
        res += a[i].v * (sr - cr * a[i].x + a[i].x * cl - sl);
    }
    printf("%lld
",res);
    
    return 0;
}

上来纠结了一会，但还是有点思路方向的。

将距离分类为该点左侧以及右侧进行处理。

每次更新一头奶牛，分别用树状数组维护记录cnt和sum，写两个upd和get。

但是代码就是想半天写不出。很棘手的问题。。。或许是因为思路不够清晰？还是要明确维护操作的细节才行啊。

欸。。。继续刷题（研究题解）吧