POJ 1990 MooFest(树状数组+离线处理)
http://poj.org/problem?id=1990
题意:
FJ有n头牛,排列成一条直线(不会在同一个点),给出每头牛在直线上的坐标x.另外,每头牛还有一个自己的声调v,如果两头牛(i和j)之间想要沟通的话,它们必须用同个音调max(v[i],v[j]),沟通起来消耗的能量为:max(v[i],v[j]) * 它们之间的距离.问要使所有的牛之间都能沟通(两两之间),总共需要消耗多少能量.
分析:
因为要求max(vi,vj),所以先读入所有的牛,将牛按它的vi值从小到大排序,所以当前处理的牛总是vi值最大的。
接下来就要算当前第i头牛的坐标xi和之前出现的所有牛的坐标之差的总和了。维护两个树状数组,数组1的sum(1,x)表示的是坐标值<=x的牛出现了多少条,数组2的sum(2,x)表示坐标值<=x的牛的坐标值总和是多少。
那么假设当前处理第i头牛,维护当前已经处理了的所有牛的坐标和total,并执行:add(1,x[i],1),add(2,x[i],x[i])
那么对于牛i来说:
它左边所有牛和它的坐标差值和为:
temp1 =sum(1,x[i])*x[i]-sum(2,x[i]).
右边的坐标差值和为:
temp2=total-sum(2,x[i])-x[i]*(i-sum(1,x[i])).
所以最终结果:ans +=(temp1+temp2)*v[i]。
结果可能超出int要用long long。
AC代码:32ms
<span style="font-size:18px;">#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=20000+100;
struct node
{
int v,x;
bool operator <(const node &b)const
{
return v<b.v;
}
}nodes[MAXN];
int c[3][MAXN];
int lowbit(int x)
{
return x&(-x);
}
int sum(int i,int x)
{
int res=0;
while(x)
{
res +=c[i][x];
x-=lowbit(x);
}
return res;
}
void add(int i,int x,int v)
{
while(x<MAXN)
{
c[i][x]+=v;
x+=lowbit(x);
}
}
int main()
{
int n;
while(scanf("%d",&n)==1&&n)
{
memset(c,0,sizeof(c));
for(int i=1;i<=n;i++)
{
scanf("%d%d",&nodes[i].v,&nodes[i].x);
}
sort(nodes+1,nodes+n+1);
int total=0;//记录所有坐标和
long long ans=0;//结果
for(int i=1;i<=n;i++)
{
int x = nodes[i].x;
total+=x;
add(1,x,1);
add(2,x,x);
int s1= sum(1,x);//在i牛前面的牛有多少头(包括i自己)
int s2= sum(2,x);//在i牛前面的牛坐标和为多少(包括i自己)
int temp1= s1*x-s2;//i左边的坐标差
int temp2= total-s2-x*(i- s1);//i右边的坐标差
ans += ((long long)temp1+temp2)*nodes[i].v;//这里不用long long强转就会溢出,直接WA
}
printf("%I64d\n",ans);
}
return 0;
}</span>