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;
}