题意:
给定 n 头牛的 听力阈值 和 所在位置,第 i 头牛和第 j 头牛相互交流所需的最小值为 max(Vi,Vj)*(Xi-Xj)。求所有牛相互交流的总和。
思路:
将牛的听力阈值降序排列。计算时当前牛的听力阈值即为最大所以只需求出 距离和牛数 就能算出当前牛到其他已经存在树状数组中牛的的总和,并更新。
这里用到两个树状数组
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn=20000;
ll c[maxn+10][2]; // 0 距离
struct node // 1 牛数
{
int v,x;
}p[maxn+10];
bool cmp(node a,node b)
{
if(a.v==b.v)
return a.x<b.x;
return a.v<b.v;
}
int lowbit(int x)
{
return x&(-x);
}
void add(int x,int v,int f)
{
while(x<=maxn)
{
c[x][f] += v;
x += lowbit(x);
}
}
ll sum(int x,int f)
{
ll ans=0;
while(x)
{
ans += c[x][f];
x -= lowbit(x);
}
return ans;
}
int main()
{
int n;
scanf("%d",&n);
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++)
scanf("%d%d",&p[i].v,&p[i].x);
sort(p+1,p+n+1,cmp);
ll a1,a2,b1,b2,ans=0;
for(int i=1;i<=n;i++)
{
a1 = sum(p[i].x,0); // 比p[i].x 距离小的牛的距离和
a2 = sum(maxn,0)-a1; // 比p[i].x 距离大的牛的距离和
b1 = sum(p[i].x,1); //在p[i].x 前边的牛数
b2 = sum(maxn,1)-b1; // 在p[i].x后边的牛数
ans += (b1*p[i].x-a1+a2-b2*p[i].x)*p[i].v;
add(p[i].x,p[i].x,0); //把距离存入树状数组
add(p[i].x,1,1); // 牛数加 1
}
printf("%lld\n",ans);
return 0;
}