POJ 1990 树状数组

非常好的一道题，算是树状数组应用的经典了。
这里我们可以很容易就感觉到这道题其实就是在求两两之间的和。每一个等于distance*max（vi,vj）。
直接n^2去做是肯定超市的，只有1000ms给了20000的数据量，所以我们需要来优化。
优化回去往如何去计算两两之间的去思考，不然一直都会有n^2的数据量。
我们将数据从小到大按照val排序，那么往后推的时候，一定有当前有的val值乘以所有和它的距离就是这个读入这个数据所带来对于和的影响。挨个往后面读入，就会解决所有的两两和的问题。
所以我们开两个数组，一个统计已经读入的，比当前读入的数据的x坐标小的数字有多少个，还有一个统计比他x小的所有数据的距离和，这样就可以计算出在当前数据前面的距离差等于个数乘以当前x减去前面的距离和，自己推算一下就能够算出来。
对于这种求前缀和的问题，最好的办法就是树状数组，最好的匹配方式就是，当点更新，区间求和，通过为何后缀和算前缀和并且用了树形结构来储存，相当的优秀啊。

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
#include<stack>
#include<vector>
#include<queue>
#include<set>
#include<map>
#define up(i,a,b)  for(int i=a;i<b;i++)
#define dw(i,a,b)  for(int i=a;i>b;i--)
#define upd(i,a,b) for(int i=a;i<=b;i++)
#define dwd(i,a,b) for(int i=a;i>=b;i--)
//#define local
typedef long long ll;
const double esp = 1e-6;
const double pi = acos(-1.0);
const long long INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int, int> pir;
int n;
struct cow {
	ll val, x;
}cows[20001];//保存输入的数据变量
ll sum[20001];//前面的距离和
ll num[20001];//后面的距离和
bool cmp(cow a, cow b)
{
	return a.val < b.val;
}//sort
int lowbit(int k)
{
	return k & (-k);
}//树状数组的坐标加减
ll querry(int k,ll *num)
{
	ll sum = 0;
	while (k>0)
	{
		sum += num[k];
		k -= lowbit(k);
	}
	return sum;
}//前面求和
void update(int i,ll val, ll *num)
{
	while (i <= 20000)
	{
		num[i] += val;
		i += lowbit(i);
	}
}//更新
int main()
{
	cin >> n;
	up(i, 0, n)
	{
		scanf("%lld %lld", &cows[i].val, &cows[i].x);
	}
	sort(cows, cows + n, cmp);
	ll ans = 0;
	up(i, 0, n)
	{//记得要用longlong才行
		ll a1 = querry(cows[i].x, num);
		ll b1 = querry(cows[i].x, sum);//因为坐标的unique，要不然需要求的是cowsx-1
		ll a2 = querry(20000, num);//算总的和，因为不一定坐标全部在左边
		ll b2 = querry(20000, sum);
	//	cout << "a1" << a1 << "b1" << b1 << "a2" << a2 << "b2" << b2 << endl;
		ans += (ll)(a1*cows[i].x - b1)*cows[i].val + (b2-b1- (a2 - a1)*cows[i].x)*cows[i].val;
		//小小的推理一下就能够得到上面的式子，其实就是左边加上右边
		//cout << ans<<endl;
		update(cows[i].x, cows[i].x, sum);//先做在更新，避免了混淆
		update(cows[i].x, 1, num);
	}
	cout << ans;
	return 0;
}