POJ 1990 MooFest(树状数组+离线处理)

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

#include
#include
#include
using namespace std;
const int MAXN=20000+100;
struct node
{
    int v,x;
    bool operator <(const node &b)const
    {
        return v