AcWing 77周赛 C题 4718. 弹球线路

原题链接

思路

每一个大主教出发点都在边上，所以可以用并查集来进行划分。因为并查集以为比较方便，所以离散化为一维，共有 $2\ast (n+m)-4$个点，最后求一下连通块。
为什么都在边上就可以则是因为两条轨迹不可能都经过同一个边上的点，证明可以去看一下y总的讲解，这个只是记录题目，附链接：y总讲解

代码

#include <bits/stdc++.h>

using namespace std;

const int N = 4000010;

int n, m;
int p[N];

int get(int i, int j)
{
    if (i == 1) return j;
    else if (j == m) return m + i - 1;
    else if (i == n) return m * 2 + n - 1 - j;
    else if (j == 1) return m * 2 + n * 2 - i - 2;
    else return 0;
}

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

int merge(int a, int b)
{
    p[find(a)] = find(b);
}

int main()
{
    cin >> n >> m;
    for (int i = 0; i <= (n + m) * 2; i ++ ) p[i] = i;
    
    if (n > m) swap(n, m);
    
    for (int i = 1; i <= m; i ++ )
    {
        if (i <= n)
        {
            merge(get(1, i), get(i, 1));
            merge(get(n, i), get(n - i + 1, 1));
        }
        else 
        {
            merge(get(1, i), get(n, i - n + 1));
            merge(get(n, i), get(1, i - n + 1));
        }
        
        if (m - i + 1 <= n)
        {
            merge(get(1, i), get(m - i + 1, m));
            merge(get(n, i), get(n - m + i, m));
        }
        else
        {
            merge(get(1, i), get(n, i + n - 1));
            merge(get(n, i), get(1, i + n - 1));
        }
    }
    
    int res = 0;
    for (int i = 1; i <= (n + m) * 2 - 4; i ++ )
        if (p[i] == i) res ++;
    
    cout << res << '\n';
    return 0;
}