【bzoj4419】[Shoi2013]发微博 STL-set

题目描述

刚开通的SH微博共有n个用户（1..n标号），在短短一个月的时间内，用户们活动频繁，共有m条按时间顺序的记录：

! x   表示用户x发了一条微博；

+ x y 表示用户x和用户y成为了好友

- x y 表示用户x和用户y解除了好友关系

当一个用户发微博的时候，所有他的好友（直接关系）都会看到他的消息。

假设最开始所有人之间都不是好友关系，记录也都是合法的（即+ x y时x和y一定不是好友，而- x y时x和y一定是好友）。

问这m条记录发生之后，每个用户分别看到了多少条消息。

输入

第1行2个整数n,m。

接下来m行，按时间顺序读入m条记录，每条记录的格式如题目所述，用空格隔开。

输出

输出一行n个用空格隔开的数（行末无空格），第i个数表示用户i最后看到了几条消息。

样例输入

2 8
! 1
! 2
+ 1 2
! 1
! 2
- 1 2
! 1
! 2

样例输出

1 1

---

题解

STL-set

考虑将一段时间内的数量转化为前缀相减的形式，即：解除关系时的总数量-建立关系时的总数量。对于每个人记录当前发的数目，然后这道题就做完了。。。

由于最后要相当于把所有关系解除，因此使用set维护建立的关系即可。

时间复杂度$O(nlog n)$

#include <set>
#include <cstdio>
#include <utility>
using namespace std;
typedef pair<int , int> pr;
set<pr> s;int cnt[200010] , ans[200010];
char str[5];
int main()
{
	int n , m , i , x , y;
	pr tmp;
	scanf("%d%d" , &n , &m);
	while(m -- )
	{
		scanf("%s%d" , str , &x);
		if(str[0] == '!') cnt[x] ++ ;
		else if(str[0] == '+') scanf("%d" , &y) , ans[x] -= cnt[y] , ans[y] -= cnt[x] , s.insert(pr(min(x , y) , max(x , y)));
		else scanf("%d" , &y) , ans[x] += cnt[y] , ans[y] += cnt[x] , s.erase(pr(min(x , y) , max(x , y)));
	}
	while(!s.empty()) tmp = *s.begin() , ans[tmp.first] += cnt[tmp.second] , ans[tmp.second] += cnt[tmp.first] , s.erase(tmp);
	for(i = 1 ; i < n ; i ++ ) printf("%d " , ans[i]);
	printf("%d
" , ans[n]);
	return 0;
}