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

 

水题，用个set维护每个人的直接朋友，记一下每个人共发了多少信息，加边删边时在ans中对应算一算就好了。

#include<cstdio>
#include<cctype>
#include<queue>
#include<cmath>
#include<cstring>
#include<set>
#include<algorithm>
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define dwn(i,s,t) for(int i=s;i>=t;i--)
#define ren for(int i=first[x];i;i=next[i])
using namespace std;
const int BufferSize=1<<16;
char buffer[BufferSize],*head,*tail;
inline char Getchar() {
    if(head==tail) {
        int l=fread(buffer,1,BufferSize,stdin);
        tail=(head=buffer)+l;
    }
    return *head++;
}
inline int read() {
    int x=0,f=1;char c=Getchar();
    for(;!isdigit(c);c=Getchar()) if(c=='-') f=-1;
    for(;isdigit(c);c=Getchar()) x=x*10+c-'0';
    return x*f;
}
const int maxn=200010;
int n,m,num[maxn],ans[maxn];
set<int>M[maxn];
int main() {
    n=read();m=read();
    while(m--) {
        char c=Getchar();
        while(c!='!'&&c!='-'&&c!='+') c=Getchar();
        if(c=='!') num[read()]++;
        else if(c=='+') {
            int x=read(),y=read();
            ans[x]-=num[y];ans[y]-=num[x];
            M[x].insert(y);M[y].insert(x);
        } 
        else {
            int x=read(),y=read();
            ans[x]+=num[y];ans[y]+=num[x];
            M[x].erase(y);M[y].erase(x);
        }
    }
    rep(i,1,n) for(set<int>::iterator it=M[i].begin();it!=M[i].end();it++)
        ans[*it]+=num[i];
    rep(i,1,n) printf("%d%c",ans[i],i==n?'
':' ');
    return 0;
}