STL
数据结构蒙蔽了我的双眼。
对每个人用一个set维护当前和他成为好友的人,并记录他发的微博数。那么对于两个人 x , y x,y x,y, x x x对 y y y的贡献即为从成为好友到解除好友中 x x x发的微博数量, y y y同理。所以减一减答案就出来了。
注意最后结束后还要算一遍。因为有些人还没有解除好友。
代码:
#include<set>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 200005
#define F inline
using namespace std;
int n,m,k,a[N],ans[N];
set <int> s[N];
set <int>::iterator it;
F char readc(){
static char buf[100000],*l=buf,*r=buf;
if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
return l==r?EOF:*l++;
}
#define pd(x) (x=='!'||x=='+'||x=='-')
F int _read(){
int x=0; char ch=readc();
while (!isdigit(ch)&&!pd(ch)) ch=readc();
if (pd(ch)) return ch;
while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
return x;
}
int main(){
for (n=_read(),m=_read();m;m--){
int f=_read(),x=_read(),y;
if (f=='!'){ a[x]++; continue; } y=_read();
if (f=='-') ans[x]+=a[y],s[x].erase(y),ans[y]+=a[x],s[y].erase(x);
else ans[x]-=a[y],s[x].insert(y),ans[y]-=a[x],s[y].insert(x);
}
for (int i=1;i<=n;i++)
for (it=s[i].begin();it!=s[i].end();it++) ans[i]+=a[*it];
for (int i=1;i<n;i++) printf("%d ",ans[i]);
return printf("%d",ans[n]),0;
}