树状数组-维护两个数组(P2345)

https://www.luogu.org/problemnew/show/P2345

开两个数组

t1用来计算坐标和

t2用来计算人数和

按v排序 可以消去max

(query2(c[i].x))*c[i].x:第i个人前面人数*第i个人的坐标

query1(c[i].x):第i个人前面人的坐标和

query1(maxn)-query1(c[i].x):第i个人后面人的坐标和

(i-query2(c[i].x))*c[i].x:第i个人后面人数*第i个人的坐标

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>

using namespace std;
const int maxn=10000005;
typedef long long ll;
ll t1[maxn],t2[maxn];
ll ans,i,n;
struct node
{
    int x,v;
}c[maxn];
ll lowbit(int i)
{
    return i&(-i);
}
void update1(ll i,ll k)
{
    while(i<=maxn)
    {
        t1[i]+=k;
        i+=lowbit(i);
    }
}
ll query1(ll i)
{
    ll sum=0;
    while(i>0)
    {
        sum+=t1[i];
        i-=lowbit(i);
    }
    return sum;
}
void update2(ll i,ll k)
{
    while(i<=maxn)
    {
        t2[i]+=k;
        i+=lowbit(i);
    }
}
ll query2(ll i)
{
    ll sum=0;
    while(i>0)
    {
        sum+=t2[i];
        i-=lowbit(i);
    }
    return sum;
}
bool cmp(node a,node b)
{
    if(a.v!=b.v)
    return a.v<b.v;
    return a.x<b.x;
}
int main()
{
    scanf("%lld",&n);
    for(int i=1;i<=n;i++)
        scanf("%d%d",&c[i].v,&c[i].x);
    sort(c+1,c+1+n,cmp);
    for(int i=1;i<=n;i++)
    {
        update1(c[i].x,c[i].x);
        update2(c[i].x,1);
        ans+=((query2(c[i].x))*c[i].x-query1(c[i].x)+query1(maxn)-query1(c[i].x)-(i-query2(c[i].x))*c[i].x)*c[i].v;
    }
    printf("%lld\n",ans);
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值