51nod 1394 差和问题(算法马拉松8）

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

解题思路

离散化+树状数组，对每次1,2操作，用树状数组要统计比它大的个数和比它小的个数

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=200000+100;
struct node
{
    int id;
    long long v;
}q[maxn];
long long a[maxn*4];
int b[maxn*4];
long long t[maxn];
long long cur[maxn];
int cou[maxn];
long long low(int k)
{
    return k&(-k);
}
void update(int k,long long v,int v2)
{
    while(k<maxn)
    {
        a[k]+=v;
        b[k]+=v2;
        k+=low(k);
    }
}
long long anss,ansn;
long long sums(int k)
{
    anss=0,ansn=0;
    while(k>0)
    {
        anss+=a[k];
        ansn+=b[k];
        k-=low(k);
    }
}
int main()
{
    int n,qu;
    while(~scanf("%d%d",&n,&qu))
    {
        for(int i=0; i<n; i++)
        {
            scanf("%I64d",&t[i]);
            cur[i]=t[i];
        }
        int cnt=n;
        for(int i=0; i<qu; i++)
        {
            scanf("%d",&q[i].id);
            if(q[i].id!=3)
            {
                scanf("%I64d",&q[i].v);
                cur[cnt]=q[i].v;
                cnt++;
            }
        }
        sort(cur,cur+cnt);
        int all=unique(cur,cur+cnt)-cur;
        long long sum=0,ans=0;
        int sumn=0;
        int temp;
        for(int i=0;i<n;i++)
        {
            temp=lower_bound(cur,cur+all,t[i])-cur+1;
            cou[temp]++;
            sumn++;
            update(temp,t[i],1);
            sum+=t[i];
            sums(temp);
            ans=ans+(t[i]*(ansn*2-sumn)-2*anss+sum);
           // cout<<ans<<endl;
        }
        long long cwt;
        for(int i=0;i<qu;i++)
        {
            if(q[i].id==1)
            {
                //cout<<q[i].id<<"  "<<q[i].v<<endl;
                temp=lower_bound(cur,cur+all,q[i].v)-cur+1;
               // cout<<temp<<endl;
                update(temp,q[i].v,1);
                cou[temp]++;
                sum+=q[i].v;
                sumn++;
                sums(temp);
               // cout<<q[i].v*(ansn*2-sumn)-2*anss+sum<<endl;
               // cout<<anss<<"  "<<ansn<<"  "<<sum<<"  "<<q[i].v<<endl;
                ans=ans+(q[i].v*(ansn*2-sumn)-2*anss+sum);
               // cout<<ans<<endl;
            }
            else if(q[i].id==2)
            {
                temp=lower_bound(cur,cur+all,q[i].v)-cur+1;
                if(cou[temp]==0)
                {
                    printf("-1\n");
                }
                else
                {
                   sums(temp);
                   ans=ans-(q[i].v*(ansn*2-sumn)-2*anss+sum);
                  update(temp,-q[i].v,-1);
                  cou[temp]--;
                  sum-=q[i].v;
                  sumn--;
                }
            }
            else if(q[i].id==3)
            {
                printf("%I64d\n",ans);
            }
        }
    }
    return 0;
}