【BZOJ5213】[ZJOI2018]迷宫（神仙题）

【BZOJ5213】[ZJOI2018]迷宫（神仙题）

题面

BZOJ
洛谷

题解

首先可以很容易的得到一个(K)个点的答案。
构建(K)个点分别表示(mod K)的余数。那么点(i)的出边(j)指向(i*m+j mod K)。容易证明这样子一定是可行的。
但是我们显然还有一部分点是可以丢掉的，即出现点等价的时候，直接合并两个点即可。
那么什么情况下两个点等价呢？显然是两个点可以到达的点集相同的时候是可以直接把这两个点给合并的。
考虑一下(i*m)在模(K)意义下相等的数的个数，令(d=gcd(m,K))，那么合法的取值有(K/d)个。定义一个参数(l)表示还有([1,l])这些数存在。如果(l>k/d)，那么在范围内可以取遍所有的合法取值，那么合并这些之后，剩下的部分递归处理，这里删去了(frac{m}{d}(k-l))个合并之后到数。否则如果(lle k/d)，或者(d=1)，证明必定两两不等，所以这(l)个数必须要。

#include<cstdio>
#include<algorithm>
using namespace std;
#define ll long long
ll m,k;int T;
ll Solve(ll l,ll k)
{
	ll d=__gcd(m,k);if(d==1||l<=k/d)return l;
	if(k<=(double)m*(k-l))return k/d;
	return m/d*(k-l)+Solve((k-m*(k-l))/d,k/d);
}
int main()
{
	scanf("%d",&T);
	while(T--)scanf("%lld%lld",&m,&k),printf("%lld
",Solve(k-1,k)+1);
	return 0;
}