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
省选竟然有这样的水题,惊了。
直接邻接表暴力加边删边能有70分,暴力就不多说了。
70分代码
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,tot=-1;
int Head[200005],ans[200005];
struct edge
{
int next,node;
bool flag;
}h[500005];
inline int read()
{
int sum=0;
char ch=getchar();
while(ch<'0'||ch>'9')
ch=getchar();
while(ch>='0'&&ch<='9')
{
sum=sum*10+ch-'0';
ch=getchar();
}
return sum;
}
void write(int x)
{
if(x>9) write(x/10);
putchar(x%10+'0');
}
inline void add(int x,int y)
{
for(register int i=Head[x];i!=-1;i=h[i].next)
{
int v=h[i].node;
if(v==y)
{
h[i].flag=1;
h[i^1].flag=1;
return ;
}
}
h[++tot].next=Head[x];
h[tot].node=y;
Head[x]=tot;
h[tot].flag=1;
h[++tot].next=Head[y];
h[tot].node=x;
Head[y]=tot;
h[tot].flag=1;
}
inline void del(int x,int y)
{
for(register int i=Head[x];i!=-1;i=h[i].next)
{
int v=h[i].node;
if(v==y)
{
h[i].flag=0;
h[i^1].flag=0;
return ;
}
}
}
inline void change(int x)
{
for(register int i=Head[x];i!=-1;i=h[i].next)
{
int v=h[i].node;
if(h[i].flag)
++ans[v];
}
}
void debug()
{
for(register int i=1;i<=n;++i)
{
for(register int j=Head[i];j!=-1;j=h[j].next)
{
cout<<h[j].node<<' ';
}
cout<<endl;
}
}
int main()
{
memset(Head,-1,sizeof(Head));
int x,y;
n=read();m=read();
char ch;
for(register int i=1;i<=m;++i)
{
ch=getchar();
if(ch=='!')
{
x=read();
change(x);
}
else if(ch=='+')
{
x=read();
y=read();
add(x,y);
}
else
{
x=read();
y=read();
del(x,y);
}
}
//debug();
for(register int i=1;i<=n;++i)
{
write(ans[i]);
putchar(' ');
}
return 0;
}
我本以为要拿满分需要用到什么数据结构来维护,结果发现直接记录后缀O(n)就能水过去了。
记录下每一次操作,从第m次操作倒着循环。用cnt[x[i]]记录下第i次操作后x发的微博的后缀和,每当碰到+时,答案加上对方的cnt。- 操作时减去对方的cnt。
100分代码
#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,tot=-1;
int x[500005],y[500005],ans[200005],cnt[200005];
char c[500005];
inline int read()
{
int sum=0;
char ch=getchar();
while(ch<'0'||ch>'9')
ch=getchar();
while(ch>='0'&&ch<='9')
{
sum=sum*10+ch-'0';
ch=getchar();
}
return sum;
}
int main()
{
n=read();m=read();
for(register int i=1;i<=m;++i)
{
c[i]=getchar();
char ch=c[i];
if(ch=='!')
x[i]=read();
else if(ch=='+')
{
x[i]=read();
y[i]=read();
}
else
{
x[i]=read();
y[i]=read();
}
}
//debug();
for(register int i=m;i;--i)
{
if(c[i]=='!')
++cnt[x[i]];
else if(c[i]=='+')
{
ans[x[i]]+=cnt[y[i]];
ans[y[i]]+=cnt[x[i]];
}
else
{
ans[x[i]]-=cnt[y[i]];
ans[y[i]]-=cnt[x[i]];
}
}
for(register int i=1;i<=n;++i)
printf("%d ",ans[i]);
return 0;
}