$O(n^2)$的暴力十分好写,现在来优化。
预处理:将牛按v从小到大排序。
需要维护两个数据:之前的牛在当前牛i左、右的牛数量以及它们的坐标。
易知i到左边所有牛的距离和=在左边的牛数×当前牛坐标-牛的坐标和。
相应地,i到左边所有牛的距离和=牛的坐标和-在右边的牛数×当前牛坐标。
开两个树状数组维护即可。
sort(a+1,a+1+n,cmp);
for(int i=1;i<=n;++i)
{
add(a[i].p,1ll,c1);add(a[i].p,a[i].p,c2);
lsum=(ll)a[i].p*ask(a[i].p-1,c1)-ask(a[i].p-1,c2);
rsum=ask(maxp,c2)-ask(a[i].p,c2)-(ll)a[i].p*(ask(maxp,c1)-ask(a[i].p,c1));
ans+=a[i].v*(lsum+rsum);
}
【POJ1990】【Luogu 5094】【USACO2004open】MooFest
原文:https://www.cnblogs.com/xzs123456/p/10427656.html