C - Peaceful Rooks (Codeforces Round #692 (Div. 2, based on Technocup 2021 Elimination Round 3)

C - Peaceful Rooks (Codeforces Round #692 (Div. 2, based on Technocup 2021 Elimination Round 3)

题意

给定 $n\times n$ 的棋盘，放入 $m$ 个车，保证 $1\le m<n$ 且没有任何的车在同一行或同一列。现在要你把所有车移动到 $(i,i)$ 位置 $(i\le 1\le n)$ ，每次移动可以在横或竖方向移动任意格，但是移动后不能和任何车处在同一列或同一行。问至少需要移动几次？

思路

最开始的方法是 $co{l}_i,ro{w}_i$ 分别代表占用第 $i$ 列的车在第 $co{l}_i$ 行，占用第 $i$ 行的车在第 $ row_i$列，然后从 $i=1$ 到 $n$ 遍历，分为行列都被占用、行被占用、列被占用。但是这样好像有点假算法，fst了（可以想象如果是现场赛有多可怕）。

后来参考了 DIV2 C. Peaceful Rooks ，用并查集做了就过了。

代码

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 2e6 + 9;
const ll MOD = 1e9 + 7;

int par[N];

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

int main()
{
    int T;
    cin >> T;
    while(T--)
    {
        int n, m;
        int cnt = 0;
        cin >> n >> m;
        for(int i = 0; i < n + 1; i++)
            par[i] = i;
        for(int i = 0, x, y; i < m; i++)
        {
            cin >> x >> y;
            if(x == y)
                continue;
            if(find(x) != find(y))
            {
                par[find(x)] = find(y);
            }
            else
            {
                cnt++;
            }
            cnt++;
        }
        cout << cnt << endl;
    }
    return 0;
}