JZOJ-senior-5946. 【NOIP2018模拟11.02】时空幻境（braid）

Time Limits: 1000 ms Memory Limits: 524288 KB

Description

Tim拥有控制时间的能力。他学会了BFS后，出了一道题：求出一张无向图中连通块的个数。他想请你做出这道题来

Input

Output

T行，每行一个数，表示联通块的个数。

Sample Input

Sample Input1
1
998244353 4
1 3

Sample Input2
5
998244353 9088
709393585 591328017
998244353 8408
476368048 122172238
998244353 217
922587543 10
998244353 2948991
40846641 7
998244353 2692315
542601916 5

Sample Output

Sample Output1
998244349

样例解释
边为{1,3},{9,27},{81,243},{729,2187} ,所以共有998244349个联通块。

Sample Output2
998235265
998235945
998244136
995295362
995552038

Data Constraint

对所有数据，n=998244353, 1<=m,k,x<998244353, 1<=T<=10000

Solution

这题和bfs没有任何关系？!
观察连边的编号，$a_1=x，a_n=x\ast k^{n-1}$
根据抽屉原理，必然是会有循环节出现的
考虑循环节长度 $len$ 的奇偶性
奇数：必然会形成一个环，联通块个数为 $n-(len-1)$
偶数：点之间仅两两连边 $n-(len/2)$
所以我们就要求出最小正整数 $a$，$k^a\equiv 1(modP)$
$a$ 一定是 $P-1$ 的约数，而 $P-1$ 又是固定的数，所以预处理它的约数，96个
对于每次询问，从小到大枚举约数，找到第一个符合的即可

Code

#include<algorithm>
#include<cstdio>

#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fd(i,a,b) for(int i=a;i>=b;--i)
#define ll long long

using namespace std;

const int M=31600,N=998244353;
int T,n,m,x,k,tot;
int g[3],f[3],a[100];

int ksm(int a,int b,int MO)
{
	int s=1;
	for(;b;a=(ll)a*(ll)a%MO,b>>=1)
		if(b&1) s=(ll)s*(ll)a%MO;
	return s;
}

void get(int t,int y)
{
	if(t>2) {a[++tot]=y; return;}
	fo(i,0,f[t])
	{
		int z=1;
		fo(j,1,i) z*=g[t];
		get(t+1,y*z);
	}
}

int main()
{
	freopen("braid.in","r",stdin);
	freopen("braid.out","w",stdout);
	g[0]=2,g[1]=7,g[2]=17;
	f[0]=23,f[1]=1,f[2]=1;
	get(0,1);
	sort(a+1,a+1+tot);
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d%d",&n,&m,&x,&k);
		int e=0,ans;
		fo(i,1,tot) if(ksm(k,a[i],n)==1) {e=a[i]; break;}
		if(e&1) ans=n-min(m,e-1);
			else ans=n-min(m,e/2);
		printf("%d\n",ans);
	}
}