最近参加了ACM的比赛,然而发现自己除了基本的C++语法和循环算法之外,其他比较方便的语法都没学过,这就很尴尬,除了签到题一道也写不出来。
在牛客网的南华大学第十五届ACM程序设计竞赛中第二题:复读机
链接:https://ac.nowcoder.com/acm/contest/699/B
来源:牛客网
在某华大学里有一个无聊的群组,群内的所有成员都是复读机,他们疯狂复读着别人的消息。然而复读机们在群内也是有阵营的,一个阵营的复读机会根据心情选择是否复读同一个阵营的成员的消息,但绝对不会去复读其他阵营的消息。
群内有n个人(标号1-n号)。现在群内聊天记录中一共有m条消息。请根据这些消息,判断出如果群内第i号成员(1<=i<=n)发一条十分有趣的消息,在同一阵营的成员都会去复读这条消息情况下,那么这条消息会被复读几次?
如果无法判断i号成员和j号成员是否为同一阵营,则视为不同阵营。
输入描述:
输入包括m+1行。第一行包括两个整数n,m;其后的m行为当前的聊天记录,每行包括一个数字i和一个字符串s(表示i发了一条消息)。(保证1<=i<=n,m<=100000,消息的字符串总长不超过100000)
输出描述:
输出一行,包括n个数字。第i个数字表示第i号成员的有趣消息会被复读几次(1<=i<=n)。数字与数字之间用一个空格分隔。
示例1
输入
3 5
1 a
2 a
1 b
2 c
3 c
输出
2 2 2
说明:
对于样例1: 由聊天记录的前两行可以得出2号和1号为同一阵营,由最后两行可得3号和2号为同一阵营,所以三者为同一阵营,他们三个发送的消息会被复读2次,故输出为2 2 2;
示例2
输入
3 3
1 a
2 b
3 c
输出
0 0 0
说明:
对于样例2:1,2,3号无法判断是否为同一阵营,视为不同阵营。
备注:
对于复读的概念:
若聊天记录为
1 a
2 a
3 a
属于2,3号复读了1号的消息。
若聊天记录为:
1 a
2 b
3 a
不构成任何复读,1的‘a‘和3的‘a’均属于自己原创消息。
从答题列的大神里扒下来的代码:
#include<bits/stdc++.h>
#define M 100005
using namespace std;
int n,m,lt;
int fa[M];
int sz[M];
int getfa(int x) { //返回根节点的值
if(fa[x]!=x)return fa[x]=getfa(fa[x]);
return x;
}
string now,lst;
int main() {
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++) {
fa[i]=i;
sz[i]=1;
}
scanf("%d",<);
cin>>lst;
m--;
while(m--) {
int a;
scanf("%d",&a);``
cin>>now;
if(now==lst) {
int x=getfa(a),y=getfa(lt);
if(x!=y){ // 如果不在一个集合
sz[x]+=sz[y]; //合并为一个集合
fa[y]=x;·//父节点变为x
}
}
lst=now;
lt=a;
}
for(int i=1; i<=n; i++)printf("%d ",sz[getfa(i)]-1);
return 0;
}
里面涉及到并查集,抽象地看是连通图的算法。具体并查集的算法理解见大佬的博客