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条记录发生之后,每个用户分别看到了多少条消息。
Input
第1行2个整数n,m。
接下来m行,按时间顺序读入m条记录,每条记录的格式如题目所述,用空格隔开。
Output
输出一行n个用空格隔开的数(行末无空格),第i个数表示用户i最后看到了几条消息。
Sample Input
2 8
! 1
! 2
+ 1 2
! 1
! 2
- 1 2
! 1
! 2
! 1
! 2
+ 1 2
! 1
! 2
- 1 2
! 1
! 2
Sample Output
1 1
只有第4和第5条记录对应的消息被看到过。其他消息发送时,1和2不是好友。
对100%的数据,N<=200000,M<=500000
Solution
感觉现在做数据结构做傻了……一看到类似操作就往数据结构上艹
后来发现唯一能搞的lct好像还搞不了emmm……
然后又想着能不能离线搞,发现+x y -x y 只有他们之间的一段才会有贡献
每个点单独开一个set,计算和其相连的人
当为叹号的时候cnt++
那么我们就在加的时候把当前cnt减掉,然后在后面减的时候再加上新的cnt,
这一段区间内的贡献就都被考虑进去了
最后再遍历一遍set把过程中没有减掉的贡献算上就行了
Code
1 #include<iostream> 2 #include<cstdio> 3 #include<set> 4 #define N (200005) 5 using namespace std; 6 set<int>s[N]; 7 set<int>::iterator po; 8 int n,m,x,y,cnt[N],ans[N]; 9 char opt[2]; 10 int main() 11 { 12 scanf("%d%d",&n,&m); 13 for (int i=1; i<=m; ++i) 14 { 15 scanf("%s",opt); 16 switch (opt[0]) 17 { 18 case '!': 19 { 20 scanf("%d",&x); 21 cnt[x]++; 22 break; 23 } 24 case '+': 25 { 26 scanf("%d%d",&x,&y); 27 s[y].insert(x); 28 s[x].insert(y); 29 ans[x]-=cnt[y]; 30 ans[y]-=cnt[x]; 31 break; 32 } 33 case '-': 34 { 35 scanf("%d%d",&x,&y); 36 s[y].erase(s[y].find(x)); 37 s[x].erase(s[x].find(y)); 38 ans[x]+=cnt[y]; 39 ans[y]+=cnt[x]; 40 break; 41 } 42 } 43 } 44 for (int i=1; i<=n; i++) 45 for (po=s[i].begin(); po!=s[i].end(); ++po) 46 ans[*po]+=cnt[i]; 47 for (int i=1; i<=n-1; ++i) 48 printf("%d ",ans[i]); 49 printf("%d",ans[n]); 50 }