RUAL 1682. Crazy Professor (并查集+预处理)

题意：

从1写到INF，如果正在写的数字b与前面的数字a存在关系：(a*a+b)%k==0 or (a+b*b)%k==0 则连一条a到b的无向边。当出现环的时候结束。

求写到多少的时候出现环。

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <list>
#include <deque>
#include <string>

#define LL long long
#define DB double
#define SI(a) scanf("%d",&a)
#define SD(a) scanf("%lf",&a)
#define SS(a) scanf("%s",a)
#define SF scanf
#define PF printf
#define MM(a,v) memset(a,v,sizeof(a))
#define REP(i,a,b) for(int (i)=(a);(i)<(b);(i)++)
#define REPD(i,a,b) for(int (i)=(a);(i)>(b);(i)--)
#define N 200009
#define INF 0x3f3f3f3f
#define EPS 1e-8
#define bug puts("bug")
using namespace std;
int n;
int fa[N];
int finfa(int k)
{
    if(fa[k]==k) return k;
    return fa[k] = finfa(fa[k]);
}
vector<int> U;
vector<int> re[N];
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("in.txt","r",stdin);
    #endif
    SI(n);
    REP(i,0,max(100,n*2)) fa[i] = i;
    REP(i,1,max(100,n*2))
    {
        int b = (n-1ll*i*i)%n;
        if(b<=0) b+=n;
        for(int j=b;j<max(100,n*2);j+=n)
        {
            if(j>i)
            re[j].push_back(i);
            if(j<i)
            re[i].push_back(j);
        }
    }
    REP(i,1,max(100,n*2))
    {
        U.clear();
        sort(re[i].begin(),re[i].end());
        REP(j,0,(int)re[i].size())
        {
            if(j&&re[i][j]==re[i][j-1]) continue;
            int k = finfa(re[i][j]);
            U.push_back(k);
            if(finfa(i)!=finfa(re[i][j]))
            fa[finfa(i)] = finfa(re[i][j]);
        }
        sort(U.begin(),U.end());
        int f = 0;
        REP(j,1,(int)U.size())
        {
            if(U[j]==U[j-1])
            {
                f = 1;
            }
        }
        if(f)
        {
            PF("%d\n",i);
            break;
        }
    }
    return 0;
}