1250. 格子游戏并查集

Posted 2021-10-25 幽殇默

https://www.acwing.com/problem/content/1252/
将二维坐标，转换成1维坐标，然后有环退出。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,p[N];
struct node
{
    int x,y;
    char op;
}temp;
vector<node>ve;
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int main(void)
{
    cin>>n>>m;
    for(int i=1;i<=n*n;i++) p[i]=i;
    for(int i=0;i<m;i++) cin>>temp.x>>temp.y>>temp.op,ve.push_back(temp);
    for(int i=0;i<m;i++)
    {
        int x=ve[i].x,y=ve[i].y;
        int xx=x,yy=y;
        if(ve[i].op=='D') xx++;
        else yy++;
        int a=(x-1)*n+y,b=(xx-1)*n+yy;
        if(find(a)!=find(b)) p[find(a)]=find(b);
        else
        {
            cout<<i+1;
            return 0;
        }
    }
    puts("draw");
    return 0;
}