格子游戏

题目描述
Alice和Bob玩了一个古老的游戏：首先画一个 n × n 的点阵（下图 n = 3）

接着，他们两个轮流在相邻的点之间画上红边和蓝边：

直到围成一个封闭的圈（面积不必为1）为止，“封圈”的那个人就是赢家。

因为棋盘实在是太大了，他们的游戏实在是太长了！他们甚至在游戏中都不知道谁赢得了游戏。

于是请你写一个程序，帮助他们计算他们是否结束了游戏？

输入格式
输入数据第一行为两个整数 n 和 m。n 表示矩阵的行数，m 表示一共画了 m 条线。
以后 m 行，每行首先有两个数字 x, y，代表了画线的起点坐标，接着用空格隔开一个字符，
假如字符是 D ，则是向下连一条边，如果是 R 就是向右连一条边。

输入数据不会有重复的边且保证正确。

输出格式
输出一行：在第几步的时候结束。
假如 m 步之后也没有结束，则输出一行 draw。

输入样例
3 5
1 1 D
1 1 R
1 2 D
2 1 R
2 2 D

输出样例
4

数据范围
n ≤ 200

---

题解
并查集：将二维坐标映射成一维坐标

#include <iostream>
using namespace std;

const int N = 40010;

int p[N];
int n, m;

int get(int x, int y)
{
    return (x - 1) * 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 = 1; i <= n * n; i ++) p[i] = i;
    
    for (int i = 1; i <= m; i ++)
    {
        char d;
        int x, y;
        cin >> x >> y >> d;
        
        int a = get(x, y), b;
        if(d == 'D') b = get(x + 1, y);
        else b = get(x, y + 1);
        
        int pa = find(a), pb = find(b);
        if(pa != pb) p[pa] = pb;
        else 
        {
            cout << i << endl;
            return 0;
        }
    }
    
    cout << "draw" << endl;
    return 0;
}