51nod1394 差和问题

1394 差和问题

有一个多重集合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

题解：

将所有数离散化下来后存进线段树，然后基本线段树操作不解释。

ps：本题不加读入优化应该是很难过去的。

代码：

#include<bits/stdc++.h>
using namespace std;
struct aaa{
	long long shu,sum,num;
}f[1000000];
struct aaaa{
	long long num,shu;
}b[1000000];
int n1,n2;
long long a[1000000],c[1000000],aa[1000000],bb[1000000],cc[1000000],d[1000000];
bool cmp(aaaa a,aaaa b){
	return a.shu<b.shu;
}
void pushup(int x){
	f[x].sum=f[x*2].sum+f[x*2+1].sum;
	f[x].num=f[x*2].num+f[x*2+1].num;
}
void xiu(int x,int l,int r,int k,int t){
	if(l==r){
		f[x].num++;
		f[x].shu=t;
		f[x].sum=f[x].num*t;
		return;
	}
	int mid=(l+r)/2;
	if(k<=mid)xiu(x*2,l,mid,k,t);
	 else xiu(x*2+1,mid+1,r,k,t);
	pushup(x); 
}
long long cha_sum(int x,int l,int r,int t,int k){
	if(t>k)return 0;
	if(l==t&&r==k)return f[x].sum;
	int mid=(l+r)/2;
	if(mid>=k)return cha_sum(x*2,l,mid,t,k);
	 else if(t>mid)return cha_sum(x*2+1,mid+1,r,t,k); 
	  else return cha_sum(x*2,l,mid,t,mid)+cha_sum(x*2+1,mid+1,r,mid+1,k);
}
long long cha_num(int x,int l,int r,int t,int k){
	if(t>k)return 0;
	if(l==t&&r==k)return f[x].num;
	int mid=(l+r)/2;
	if(mid>=k)return cha_num(x*2,l,mid,t,k);
	 else if(t>mid)return cha_num(x*2+1,mid+1,r,t,k); 
	  else return cha_num(x*2,l,mid,t,mid)+cha_num(x*2+1,mid+1,r,mid+1,k);
}
void del(int x,int l,int r,int t){
	if(l==r){
		f[x].num--;
		f[x].sum-=f[x].shu;
		return;
	}
	int mid=(l+r)/2;
	if(t<=mid)del(x*2,l,mid,t);
	 else del(x*2+1,mid+1,r,t);
	pushup(x);
}
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int main(){
	int n,m,i;
	long long ans=0,t;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;i++){
		a[i]=read();
		n1++;
		b[n1].shu=a[i];
		b[n1].num=i;
	}
		for(i=1;i<=m;i++){
		aa[i]=read();
		if(aa[i]!=3){
		 	bb[i]=read();
		 	n1++;
		 	b[n1].shu=bb[i];
		 	b[n1].num=i+n;
		 }
	}
	sort(b+1,b+n1+1,cmp);
	b[0].shu=-1;
	for(i=1;i<=n1;i++){
		if(b[i].shu!=b[i-1].shu){
			n2++;
			c[n2]=b[i].shu;
			d[n2]=1;
		}
		 else d[n2]++;
		cc[b[i].num]=n2; 	 
	}
	for(i=1;i<=n;i++){
		ans+=cha_num(1,1,n2,1,cc[i]-1)*a[i]-cha_sum(1,1,n2,1,cc[i]-1)+cha_sum(1,1,n2,cc[i],n2)-cha_num(1,1,n2,cc[i],n2)*a[i];
		xiu(1,1,n2,cc[i],a[i]);
	}
	for(i=1;i<=m;i++){
		if(aa[i]==3)printf("%lld\n",ans);
		 else{
		 	if(aa[i]==1){
		 		ans+=cha_num(1,1,n2,1,cc[i+n]-1)*bb[i]-cha_sum(1,1,n2,1,cc[i+n]-1)+cha_sum(1,1,n2,cc[i+n]+1,n2)-cha_num(1,1,n2,cc[i+n]+1,n2)*bb[i];
		 		xiu(1,1,n2,cc[i+n],bb[i]);
			 }
			  else{
			  	t=cha_num(1,1,n2,cc[i+n],cc[i+n]);
			  	if(!t)printf("-1\n");
			  	 else{
			  	 	ans-=cha_num(1,1,n2,1,cc[i+n]-1)*bb[i]-cha_sum(1,1,n2,1,cc[i+n]-1)+cha_sum(1,1,n2,cc[i+n]+1,n2)-cha_num(1,1,n2,cc[i+n]+1,n2)*bb[i];
			  	 	del(1,1,n2,cc[i+n]);
				   }
			  }
		 }
	}
}