BZOJ4419: [Shoi2013]发微博（洛谷P3998）

STL

BZOJ题目传送门
洛谷题目传送门

数据结构蒙蔽了我的双眼。

对每个人用一个set维护当前和他成为好友的人，并记录他发的微博数。那么对于两个人$x,y$，$x$对$y$的贡献即为从成为好友到解除好友中$x$发的微博数量，$y$同理。所以减一减答案就出来了。

注意最后结束后还要算一遍。因为有些人还没有解除好友。

代码：

#include
#include
#include
#include
#include
#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;
}