题意:
从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;
}