51nod 1394：差和问题

1394 差和问题

基准时间限制：1 秒 空间限制：131072 KB 分值: 80  难度：5级算法题

 收藏

 关注

有一个集合S，初始状态下有n个元素，对他进行如下操作：

1、向S里面添加一个值为v的元素。输入格式为1 v

2、向S里面删除一个值为v的元素。输入格式为2 v

3、询问S里面的元素两两之差绝对值之和。输入格式为3

 

对于样例，

操作3，|1-2|+|1-3|+|2-3|=4

操作1 4之后，集合中的数字为1 2 3 4

操作3，|1-2|+|1-3|+|2-3|+|1-4|+|2-4|+|3-4|=10

操作2 2之后，集合中的数字为1 3 4

操作3，|1-3|+|1-4|+|3-4|=6

Input

第一行输入两个整数n，Q表示集合中初始元素个数和操作次数。(1<=n,Q<=100,000)
第二行给出n个整数a[0],a[1],a[2],…,a[n-1]，表示初始集合中的元素。(0<=a[i]<=1,000,000,000) 
接下来Q行，每行一个操作。(0<=v<=1,000,000,000)

Output

对于第2类操作，如果集合中不存在值为v的元素可供删除，输出-1。
对于第3类操作，输出答案。

Input示例

3 5
1 2 3
3
1 4
3
2 2
3

Output示例

4
10
6

先思考一个简单的问题，给定N个数字，求两两之间差的绝对值之和。
这个很明显先排个序，这样就可以把绝对值去掉了。然后从小到大统计每一个和其它组成的差绝对值之和。
再联系到本题中。N+Q<=2e5,那么先把所有的数据读入，最多有2e5种数字，离散化一下，对应到树状数组之中。插入和删除一个数字的时候要统计一下这个对答案的影响。
当前数字为x,比当前数字小的有cnt个，总和为sum,那么这一部分对答案的影响是x*cnt-sum.对于求比当前数字大的数字类似的道理。
总的复杂度是(n+Q)log(n+Q)。

以后记住了，这种增删查的树状数组很方便。。。另外，Visual C++提交~

代码：

#pragma warning(disable:4996)  
#include <iostream>  
#include <algorithm>  
#include <cmath>  
#include <vector>  
#include <string>  
#include <cstring>
#include <queue>
#include <map>
using namespace std;
typedef long long ll;

const int maxn = 100005;
ll N;
ll n, q;
struct o
{
	ll op;
	ll num;
}oper[maxn];

map<ll, ll>ma;
ll val[2 * maxn], val2[2 * maxn];
ll ans[2 * maxn], nnn[2 * maxn];
ll exsit[2 * maxn];
ll res_wc;

ll lowbit(ll x)
{
	return x&(-x);
}

void add(ll x,ll y)
{
	while (x <= N)
	{
		ans[x] += y;
		x = x + lowbit(x);
	}
}
void add2(ll v,ll x, ll y)
{
	while (x <= N)
	{
		exsit[x] += v*y;
		x = x + lowbit(x);
	}
}

ll sum(ll x)
{
	ll res = 0;
	while (x>0)
	{
		res += ans[x];
		x = x - lowbit(x);
	}
	return res;
}

ll sum2(ll x)
{
	ll res = 0;
	while (x>0)
	{
		res += exsit[x];
		x = x - lowbit(x);
	}
	return res;
}

int main()
{
	//freopen("i.txt", "r", stdin);
	//freopen("o.txt", "w", stdout);
	
	ll i, j, k;
	scanf("%lld%lld", &n, &q);
	
	for (i = 1; i <= n; i++)
	{
		scanf("%lld", &val[i]);
		val2[i] = val[i];
	}
	j = n;
	for (i = 1; i <= q; i++)
	{
		scanf("%lld", &oper[i].op);
		if (oper[i].op == 1 || oper[i].op == 2)
		{
			scanf("%lld", &oper[i].num);
			j++;
			val2[j] = oper[i].num;
		}
	}
	N = j;
	sort(val + 1, val + n + 1);
	sort(val2 + 1, val2 + j + 1);
	k = 0;
	for (i = 1; i <= j; i++)
	{
		if (ma[val2[i]])continue;
		ma[val2[i]]= ++k;
	}
	res_wc = 0;
	ll temp, temp_a, temp_b, temp_before_sum, temp_after_sum;
	ll su = 0;//所有元素的和
	for (i = 1; i <= n; i++)
	{
		ll x = ma[val[i]];
		add(x, 1);
		add2(val[i], x, 1);
		temp = sum(x - 1);//求当前位置的前面有多少个元素
		res_wc += temp*val[i] - (su-(nnn[x]*val[i]));
		nnn[x]++;
		su = su + val[i];
	}
	for (i = 1; i <= q; i++)
	{
		if (oper[i].op == 3)
		{
			printf("%lld\n", res_wc);
		}
		else if (oper[i].op == 2)
		{
			ll xx = ma[oper[i].num];
			if (nnn[xx]==0)
			{
				printf("-1\n");
			}
			else
			{
				su = su - oper[i].num;
				add(xx, -1);
				add2(oper[i].num, xx, -1);
				temp_b = sum(xx - 1);//前面有多少个元素
				temp_a = n - temp_b - nnn[xx];//n表示当前元素个数
				n--; nnn[xx]--;
				temp_before_sum = sum2(xx - 1);//前面有多少个和
				temp_after_sum = su - temp_before_sum - nnn[xx] * oper[i].num;//后面元素的和

				res_wc = res_wc + (temp_before_sum - 1LL*oper[i].num*temp_b) + (1LL*temp_a*oper[i].num - temp_after_sum);
			}
		}
		else
		{
			ll xx = ma[oper[i].num];
			su = su + oper[i].num;
		
			add(xx, 1);
			add2(oper[i].num, xx, 1);
			temp_b = sum(xx - 1);//前面有多少个元素
			temp_a = n - temp_b - nnn[xx];//n表示当前元素个数
			nnn[xx]++;
			n++;
			temp_before_sum = sum2(xx - 1);//前面有多少个和
			temp_after_sum = su - temp_before_sum - nnn[xx] * oper[i].num;//后面元素的和

			res_wc = res_wc - (temp_before_sum - 1LL*oper[i].num*temp_b) - (1LL*temp_a*oper[i].num - temp_after_sum);
		}
	}
	//system("pause");
	return 0;
}