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
Sample Output
1 1
只有第4和第5条记录对应的消息被看到过。其他消息发送时,1和2不是好友。
对100%的数据,N<=200000,M<=500000
想说的话
没有
正解
这个题我们看出来是在维护任意两个人之间的关系
(是好友,不是好友 包括看到消息也建立在好友的基础上)
所以很容易想到map的映射关系
用cnt数组存每个人现在的微博数
然后好友神马的用map表示
成为了好友映射的时候就加上微博数量
删除了之后就减掉
最后用一个迭代器扫一遍就水过去了
我是第一次写map
写的特别丑qwq
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<map>
using namespace std;
#define N 200001
#define M 500001
int ans[N],cnt[N],n,m;
map<int,int>mp[M];
int main()
{
scanf("%d%d",&n,&m);
while(m--)
{
char ch=getchar();
while(ch!='!'&&ch!='+'&&ch!='-') ch=getchar();
if(ch=='!')
{
int x;
scanf("%d",&x);
cnt[x]++;
}
else if(ch=='+')
{
int x,y;
scanf("%d%d",&x,&y);
mp[x][y]=-cnt[x];
mp[y][x]=-cnt[y];
}
else if(ch=='-')
{
int x,y;scanf("%d%d",&x,&y);
ans[y]+=cnt[x]+mp[x][y];
ans[x]+=cnt[y]+mp[y][x];
mp[x].erase(y);mp[y].erase(x);
}
}
for(int i=1;i<=n;i++)
{
map<int , int>::iterator t;
for (t=mp[i].begin();t!=mp[i].end();++t)
ans[t->first]+=t->second+cnt[i];
}
cout<<ans[1];
for(int i=2;i<=n;i++) printf(" %d",ans[i]);
cout<<endl;
}
写在最后
其实map这个stl用来映射挺好用的
每次操作都是log级别
而且是为数不多stl时间比较良心的东西