[文章目录]
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条记录发生之后，每个用户分别看到了多少条消息。N<=200000，M<=500000
感看到题目的时候挺吓人的。
仔细想一下每个操作对答案的贡献:
解除好友：对答案的贡献为当前微博数-加好友的时候的微博数
所以直接用set维护好友的集合，加的时候减去微博数，解除的时候加上微博数，最后在统计一遍依然是好友的微博数就好了。
#include <set>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 201000
int n,m;
int ans[N],now[N];
set <int>s[N];
set <int>::iterator it;
char st[20];
int main()
{
scanf("%d%d",&n,&m);
int x,y;
while(m--)
{
scanf("%s",st);
if(st[0]=='!')
{
scanf("%d",&x);
now[x]++;
}
else if(st[0]=='-')
{
scanf("%d%d",&x,&y);
s[x].erase(y); s[y].erase(x);
ans[x]+=now[y];
ans[y]+=now[x];
}
else
{
scanf("%d%d",&x,&y);
s[x].insert(y);
s[y].insert(x);
ans[x]-=now[y];
ans[y]-=now[x];
}
}
for(int i=1;i<=n;i++)
{
it=s[i].begin();
while(it!=s[i].end())
{
ans[*it]+=now[i];
it++;
}
}
for(int i=1;i<=n;i++)
printf("%d%c",ans[i],(i==n)?'\n':' ');
return 0;
}