UOJ#375. 【ZJOI2018】迷宫
原文链接https://www.cnblogs.com/zhouzhendong/p/UOJ375.html
题解
首先，我们可以建出一个 k 个点的自动机，第 i 个点表示当前数对 k 取模为 i-1。显然每一个点有 m-1 条出边。
然后，稍加观察，我们就可以发现，如果一些节点的出边集合是相同的，我们就可以将他们合并。
具体是怎样的节点呢？
对于每一个 $t$ ，满足
$$im \equiv t\pmod k $$
的所有 $i$ 都可以合并（节点 0 除外）。
然后你就发现你几乎过了大样例的前 100 个数据。
只有很少是错的。
所以大概会得到0分的好成绩。
那么错在了哪儿？
并不是这样合并完了就没有节点可以合并了。
我们再想想哪些节点可以合并。
注意到如果 $im^2\equiv jm^2\pmod k $ 的话，只要在后面接上两个字母，$i$ 和 $j$ 就没区别了。
但是如果其中一个走一步就可以到 0 了，另一个不行，那么他们就是不等价的。
除此之外的情况都可以合并。
类似地，对于 $m^3,m^4,m^5,\cdots$ 也差不多。
于是我们就可以得到一个 $O(k\log ^2 k)$ 的模拟。
可以获得 50 分的好成绩。
再看看质数的情况，由于当 $\gcd(k,m)=1$ 时，答案等于 $k$ ，所以 $k$ 一定是 $m$ 的倍数。稍微模拟一下可以发现，一次合并之后相当于缩小了问题规模，而去求 $k'=k/m,m'=m$ 的子问题。简单推导一下就可以得到质数的10分。
现在已经得到了 60 分的好成绩。
现在我们来看看 60 分的做法给了我们什么启发：缩小问题规模。
设 $d = \gcd(k,m)>1$ ，则一开始给 $1\cdots k-1$ （0是特殊点被ban掉了）乘了 m 模了 k 之后，所有数都是 d 的倍数。
设 $t=\frac kd$ ，则 $1\cdots k-1$ 乘m模k 后的结果是 $t$ 个一循环的，所以 d 的每一个倍数都会出现。我们把每一种乘m模k相同的缩起来显然是最优的，那么，我们可以合并 $(k-1) - t$ 次。
现在还剩下 $t$ 个节点，注意到现在加上 $0\cdots m-1$ 之后与 k 同余的节点显然不能参与下一次合并，所以我们要把他们扔掉。
扔掉的是那些呢？
好像有点棘手。我们先绕个弯。之前说道这里的所有数都是 d 的倍数。所以我们把所有值都除以 d，k' = k/d, m'=m/d 。由于之前提到 " d 的每一个倍数都会出现"，所以所有值除以 d 之后，得到的就是 $1,2,\cdots , k'-1$ 。我们扔掉后 $m'$ 个数，所以剩下的就是 $1,2,\cdots ,k'-m$ 。
我们发现我们得到了一个原问题的子问题。
于是我们可以考虑把这种问题一般化：
假设读入的两个数第一个是 n ，第二个是 k 。
假设当前剩余节点 $1,2,\cdots ub$ ，$m$ 为 $n$ 的若干次幂除以一些 gcd ，$k$ 表示除过一些 gcd 的 k ，在这种状态下，我们来计算最多可以合并多少节点。初始情况下， ub = k - 1, m = 1, k = k 。
令 $d = \gcd(k,n),t=k/d$ ，
考虑分几种情况讨论：
1. d = 1： 显然不能合并。
2. $ub\leq t$： 乘以 m 之后的得到的结果互不相同，所以也不能合并。
3. 这种情况下可以合并 $ub - t$ 次。合并完之后，显然还剩 $t$ 个点。这时候 $m' = m\cdot (n/d)$，扔掉 $m'$ 个点之后还剩 $t-m'$ 个，所以 $ub' = t - m'$ ，剩下就是 $k' = t$ 。
注意一下运算的时候不要爆 long long 。
于是你就可以得到 100 分的好成绩。
代码1 - 60分
#include <bits/stdc++.h>
#define clr(x) memset(x,0,sizeof (x))
#define For(i,a,b) for (int i=a;i<=b;i++)
#define Fod(i,b,a) for (int i=b;i>=a;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define _SEED_ ('C'+'L'+'Y'+'A'+'K'+'I'+'O'+'I')
#define outval(x) printf(#x" = %d\n",x)
#define outvec(x) printf("vec "#x" = ");for (auto _v : x)printf("%d ",_v);puts("")
#define outtag(x) puts("----------"#x"----------")
using namespace std;
typedef long long LL;
LL read(){
LL x=0,f=0;
char ch=getchar();
while (!isdigit(ch))
f|=ch=='-',ch=getchar();
while (isdigit(ch))
x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return f?-x:x;
}
LL T,n,k;
LL gcd(LL a,LL b){
return b?gcd(b,a%b):a;
}
vector <int> v;
int tn(int x){
return x?x:k;
}
int main(){
T=read();
while (T--){
n=read(),k=read();
if (gcd(n,k)==1){
printf("%lld\n",k);
continue;
}
if (n>1e5||k>1e5){
LL ans=k,c=0;
while (ans%n==0)
ans/=n,c++;
printf("%lld\n",ans+c);
continue;
}
LL m=1,ans=k;
v.clear();
For(i,0,k-1)
v.pb(tn(i));
while (1){
sort(v.begin(),v.end());
// outvec(v);
if (v.empty())
break;
int cnt=unique(v.begin(),v.end())-v.begin();
while (v.size()>cnt)
v.pop_back(),ans--;
// outvec(v);
while (!v.empty()&&v.back()>k-m)
v.pop_back();
// outvec(v);
for (auto &i : v)
i=tn(i*n%k);
if (k/m==0)
break;
m*=n;
}
cout<<ans<<endl;
}
return 0;
}
代码2 - 100分
#include <bits/stdc++.h>
#define clr(x) memset(x,0,sizeof (x))
#define For(i,a,b) for (int i=a;i<=b;i++)
#define Fod(i,b,a) for (int i=b;i>=a;i--)
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define _SEED_ ('C'+'L'+'Y'+'A'+'K'+'I'+'O'+'I')
#define outval(x) printf(#x" = %d\n",x)
#define outvec(x) printf("vec "#x" = ");for (auto _v : x)printf("%d ",_v);puts("")
#define outtag(x) puts("----------"#x"----------")
using namespace std;
typedef long long LL;
LL read(){
LL x=0,f=0;
char ch=getchar();
while (!isdigit(ch))
f|=ch=='-',ch=getchar();
while (isdigit(ch))
x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return f?-x:x;
}
LL T,n,k;
LL gcd(LL a,LL b){
return b?gcd(b,a%b):a;
}
int main(){
T=read();
while (T--){
n=read(),k=read();
LL ans=k,ub=k-1,m=1;
while (ub>0){
LL d=gcd(k,n),t=k/d;
if (d==1||ub<=t)
break;
ans-=ub-t;
if (t/m/(n/d)==0)
break;
m*=n/d;
ub=t-m;
k/=d;
}
printf("%lld\n",ans);
}
return 0;
}
