Alice和Bob玩了一个古老的游戏:首先画一个 n×n 的点阵(下图 n=3 )。
接着,他们两个轮流在相邻的点之间画上红边和蓝边:
直到围成一个封闭的圈(面积不必为 1)为止,“封圈”的那个人就是赢家。因为棋盘实在是太大了,他们的游戏实在是太长了!
他们甚至在游戏中都不知道谁赢得了游戏。
于是请你写一个程序,帮助他们计算他们是否结束了游戏?
输入格式
输入数据第一行为两个整数 n 和 m。n表示点阵的大小,m 表示一共画了 m 条线。
以后 m 行,每行首先有两个数字 (x,y),代表了画线的起点坐标,接着用空格隔开一个字符,假如字符是 D,则是向下连一条边,如果是 R 就是向右连一条边。
输入数据不会有重复的边且保证正确。
输出格式
输出一行:在第几步的时候结束。
假如 m 步之后也没有结束,则输出一行“draw”。
数据范围
1≤n≤200,
1≤m≤24000
输入样例:
3 5
1 1 D
1 1 R
1 2 D
2 1 R
2 2 D
输出样例:
4
算法:一维数组并查集。
核心算法:先明确在并查集中,每个集合都有一个头,初始化时,集合的头就是他自己。
1(初始化)
int a[50000];//只要大于200x200即可。
int d,b,n,m,i;//d,b为进行D操作时的行列坐标,n为矩阵大小,m为操作次数
char c;
for( i=1;i<=n*n;i++)//为n*n列的每一个结点赋值,初值等于它的下标,也就是每个结点的头就是他自己。
a[i]=i;
2(找头结点的函数,*并查集中的精华所在)
int find(int x){
if(a[x]!=x)a[x]=find(a[x]) ;//如果a[x]不等于x,也就是a[x]的头不是他自己,递归查找他的头,并缩短路径,用a[x]的头给a[x]赋值。
return a[x];//返回a[x]//返回的是a[x]的头。
}
3.在进行D,R操作前,先判断是否还差一步就可以结束游戏,如果进行相应的D,R操作后,游戏结束,输出循环次数i。
for( i=1;i<=m;i++){
int x1;//x1储存当前操作在矩阵中的下标。
cin>>d>>b>>c;
x1=(d-1)*n+b;//用一维数组表示当前操作下标,x1+n表示下一列,x1+1表示下一个坐标。
if(find(x1)==find(x1+n)&&c=='D'){//判断在经行D操作前x1和即将要操作的x1+n这个集合的头是否是同一个头,如果是则输出当前循环次数i,结束游戏。
cout<<i<<endl;
return 0;
}
else if(find(x1)==find(x1+1)&&c=='R'){//同理
cout<<i<<endl;
return 0;
}
if(c=='D'){//进行相应的D,R操作。
a[find(x1)]=find(x1+n);//将x1的头和x1+n的头连接起来成为一个集合,因为是要给下标为x1的头的集合赋值,所以是a[find(x1],不能是find(x1)=find(x1+n)。
}
else
a[find(x1)]=find(x1+1);//同理
}
时间复杂度
缩短路径后近似于o(1)。
参考文献
bilibili UP主:哩风
C++ 代码
#include<iostream>
using namespace std;
int a[50000];
int find(int x){
if(a[x]!=x)a[x]=find(a[x]) ;
return a[x];
}
int main(){
int d,b,n,m,i;
char c;
cin>>n>>m;
for( i=1;i<=n*n;i++)
a[i]=i;
for( i=1;i<=m;i++){
int x1,x2;
cin>>d>>b>>c;
x1=(d-1)*n+b;
if(find(x1)==find(x1+n)&&c=='D'){
cout<<i<<endl;
return 0;
}
else if(find(x1)==find(x1+1)&&c=='R'){
cout<<i<<endl;
return 0;
}
if(c=='D'){
a[find(x1)]=find(x1+n);
}
else
a[find(x1)]=find(x1+1);
}cout<<"draw";
}