2019.03.04【ZJOI2018】【BZOJ5213】【洛谷P4339】迷宫（自动机）（数论）

BZOJ传送门

洛谷传送门

---

解析：

这道题本质其实是要最小化一个能够接受所有$m$进制下$K$的倍数的自动机。

显然我们有一个方案就是$K$个节点，$i$的第$j$条边向$(i\ast m+j)\%K$连过去。

所以我们有平凡的上界$K$，现在考虑合并自动机节点。

两个节点是等价的，当且仅当它们的所有能够接受的后继转移相同。

换句话说，设$f(a)$是$a$状态转移到$0$的最小步数，如果$a,b$等价，则$f(a)=f(b)$且
$a\ast m^{f(a)}\equiv b\ast m^{f(b)}(\mathrm{m}\mathrm{o}\mathrm{d}K)$，反之也成立。

显然$0$是一个单独的等价类，同时是开始状态和终止状态。

现在考虑去掉等价的状态，将所有数直接乘上$m$看落在$\%K$的哪个等价类中。这时候$[K-m+1,K]$中的数就是$f(i)=1$的，令$d=\gcd (K,m)$，此时所有数都是$d$的倍数。

显然，如果$d=1$，所有数对应不同的等价类，没有可以删去的数，不然就将$[K-m+1,K]$中的数去重并加入答案。

然后再将剩下的数$\ast m$，将$[K-m^2+1,K]$中的数去重并加入答案。

令$solve(L,m,K)$表示保留$1-L$中的数的时候的答案。

如果$L\le K/d$，所有数都会对应不同的值，没有等价的。

否则，$im\%K$取遍了$K/d$的所有值，剩下$[1,K-m(K-L)]$中$d$的倍数，递归处理就好了。

注意如果直接递归处理，$m\ast (K-L)$的乘法会爆$longlong$，所以先判断一下区间内有没有数再继续做。

---

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	inline char get_char(){
		static cs int Rlen=1<<20|1;
		static char buf[Rlen],*p1,*p2;
		return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
	}
	
	inline ll getint(){
		re char c;
		while(!isdigit(c=gc()));re ll num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
}
using namespace IO;

inline ll solve(ll L,ll m,ll K){
	ll d;
	if((d=__gcd(m,K))==1)return L;
	if(L<=K/d)return L;
	if(K/(K-L)>=m)return m/d*(K-L)+solve(K/d-m/d*(K-L),m,K/d);
	else return K/d;
}

int T;
ll m,K;
signed main(){
	T=getint();
	while(T--){
		m=getint(),K=getint();
		cout<<solve(K-1,m,K)+1<<"\n";
	}
	return 0;
}