BZOJ4419:[SHOI2013]发微博(乱搞)

Description

刚开通的SH微博共有n个用户（1..n标号），在短短一个月的时间内，用户们活动频繁，共有m条按时间顺序的记录：

! x   表示用户x发了一条微博；

+ x y 表示用户x和用户y成为了好友

- x y 表示用户x和用户y解除了好友关系

当一个用户发微博的时候，所有他的好友（直接关系）都会看到他的消息。

假设最开始所有人之间都不是好友关系，记录也都是合法的（即+ x y时x和y一定不是好友，而- x y时x和y一定是好友）。

问这m条记录发生之后，每个用户分别看到了多少条消息。

Input

第1行2个整数n,m。

接下来m行，按时间顺序读入m条记录，每条记录的格式如题目所述，用空格隔开。

Output

输出一行n个用空格隔开的数（行末无空格），第i个数表示用户i最后看到了几条消息。

Sample Input

2 8
! 1
! 2
+ 1 2
! 1
! 2
- 1 2
! 1
! 2

Sample Output

1 1
只有第4和第5条记录对应的消息被看到过。其他消息发送时，1和2不是好友。

对100%的数据，N<=200000，M<=500000

Solution

感觉现在做数据结构做傻了……一看到类似操作就往数据结构上艹
后来发现唯一能搞的lct好像还搞不了emmm……
然后又想着能不能离线搞，发现+x y  -x y 只有他们之间的一段才会有贡献
每个点单独开一个set，计算和其相连的人
当为叹号的时候cnt++
那么我们就在加的时候把当前cnt减掉，然后在后面减的时候再加上新的cnt，
这一段区间内的贡献就都被考虑进去了
最后再遍历一遍set把过程中没有减掉的贡献算上就行了

Code

 #include<iostream>
 #include<cstdio>
 #include<set>
 #define N (200005)
 using namespace std;
 set<int>s[N];
 set<int>::iterator po;
 int n,m,x,y,cnt[N],ans[N];
 char opt[];
 int main()
 {
     scanf("%d%d",&n,&m);
     for (int i=; i<=m; ++i)
     {
         scanf("%s",opt);
         switch (opt[])
         {
             case '!':
             {
                 scanf("%d",&x);
                 cnt[x]++;
                 break;
             }
             case '+':
             {
                 scanf("%d%d",&x,&y);
                 s[y].insert(x);
                 s[x].insert(y);
                 ans[x]-=cnt[y];
                 ans[y]-=cnt[x];
                 break;
             }
             case '-':
             {
                 scanf("%d%d",&x,&y);
                 s[y].erase(s[y].find(x));
                 s[x].erase(s[x].find(y));
                 ans[x]+=cnt[y];
                 ans[y]+=cnt[x];
                 break;
             }
         }
     }
     for (int i=; i<=n; i++)
         for (po=s[i].begin(); po!=s[i].end(); ++po)
             ans[*po]+=cnt[i];
     for (int i=; i<=n-; ++i)
         printf("%d ",ans[i]);
     printf("%d",ans[n]);
 }