题目描述
刚开通的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(n\log 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\n" , ans[n]);
return 0;
}