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
思路:
这道题可以用并查集来写,每个要保存的信息是一个二维的点,二维变一维,用并查集维护,当我们增加的这条边的两个点已经在一个集合中,那么就连成环了。
代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 40010;
int n,m;
int p[N];
int get(int x,int y){
return x*n+y;
}
int find(int x){
if(p[x]!=x)p[x]=find(p[x]);
return p[x];
}
int main(){
cin>>n>>m;
for(int i=0;i< n*n;i++)p[i]=i;
int res = 0;
for(int i=0;i<m;i++){
int x,y;
char d;
cin>>x>>y>>d;
x--,y--;
int a = get(x,y),b;
if(d == 'D')b = get(x+1,y);
else b = get(x,y+1);
int fa = find(a),fb = find(b);
if(fa==fb){
res = i+1;
break;
}else{
p[fa] = fb;
}
}
if(!res)puts("draw");
else cout<<res;
}