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[2];
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1; i<=m; ++i)
    {
        scanf("%s",opt);
        switch (opt[0])
        {
            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=1; i<=n; i++)
        for (po=s[i].begin(); po!=s[i].end(); ++po)
            ans[*po]+=cnt[i];
    for (int i=1; i<=n-1; ++i)
        printf("%d ",ans[i]);
    printf("%d",ans[n]);
}