AcWing 1250. 格子游戏

AcWing 1250. 格子游戏

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

这道题我在想怎么把二维的数组变成一维的，因为我一开始就有思路，然后y总确实和我方法差不多，把二维的x,y变成一维的且不重复的方法是先(x–,y–)然后xn+y，这样就变成了(n-1)(n-1)的数组，这个数组刚好有一个性质。

对于上图来说，每一个格子里的数与它的坐标的关系是
x=g/n y=g%n

根据这个式子反推可以得到

g=x*n+y

我当时看完y总才明白过来，思路就出来了，把每一条边放到一个集合上，如果一条边的两个点都在一个集合，说明存在环。

#include<iostream>
#include<algorithm>

using namespace std;

const int N=210*210;

int p[N];
int n,m;

int find(int x)
{
    if(p[x]==x) return p[x];
    return p[x]=find(p[x]);
}

int get(int a,int b)
{
    return a*n+b;
}
int main(void)
{
    cin>>n>>m;
    for(int i=0;i<n*n;i++) p[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        char op;
        cin>>x>>y>>op;
        x--;y--;
        int pa=get(x,y);
        int pb;
        if(op=='D') pb=get(x+1,y);
        else
        pb=get(x,y+1);
        pa=find(pa),pb=find(pb);
        if(pa==pb)
        {
            cout<<i;
            return 0;
        }
        else
        p[pa]=pb;
    }
    printf("draw");
}