AtCoder Beginner Contest 290 D - Marking (结合官方题解的证明和解析) 思维 + 裴蜀定理

D - Marking

题目链接：D - Marking
官方题解链接：Editorial

题意

给定 $N,D,K$
经过的每个格子都会被粉刷 从$idx=0$出发(第一次粉刷第$0$格)
1. 每次向前跳$D$步$(idx+D)\mathrm{mod}N$，若该格子未被粉刷则粉刷，继续执行第$1$步。
2. 若之前被粉刷过则向前一直走$(idx+1)\mathrm{mod}N$到没粉刷的格子为止，粉刷该格子后回到第$1$步。
问第$K$次粉刷的是哪个格子。

前置知识 （裴蜀定理有关）

首先有 结论： 设 $g=gcd(A,B)$ $A=ag$，$B=bg$。
$a$项式子 $0B\mathrm{mod}A$， $B\mathrm{mod}A$， $2B\mathrm{mod}A$，……，$(a-1)B\mathrm{mod}A$。
1.每项式子的值都是$g$的倍数 。
2.且倍数为$0\sim a-1$每一个数恰好出现一次。

证明：

1. $B\mathrm{mod}A=B-xA(x=B/A)=bg-xag=(b-xa)g$。
2. 设有重复的倍数 $iB\mathrm{mod}A=jB\mathrm{mod}A(i!=j)$。
   $iB-jB=(i-j)bg=yA$ 为 $A$ 的倍数 $(iB-jB)\mathrm{mod}A=0$ 模 $A$ 余数相同，那么相减后再模 $A$ 余数自然为 $0$。
   $(i-j)bg=yag$
   因为 $a$ 与 $b$ 互质( $a$ 与 $b$ 若还有大于 $1$ 的公约数，那么 $gcd$ 就不会是 $g$ 而是 $g\ast gcd(a,b)$)。
   所以 $(i-j)$ 是 $a$ 的倍数 这显然与上式系数区间 $[0,a-1]$相违背。

解题思路

设 $g=gcd(N,D)$ $N=ng$，$D=dg$。
对于上述问题有跳跃的前 $n$ 步为 $0\mathrm{mod}N$， $D\mathrm{mod}N$，$2D\mathrm{mod}N$，…… ，$(n-1)D\mathrm{mod}N$。
每一项都是不重复的 $g$ 的倍数，所以不会触发条件 $2$。 而第 $n+1$ 步 $nD\mathrm{mod}N=ndg=dN\mathrm{mod}N=0$。
开始有重复项第二轮开始，此时需要向前走一步 $(0$，接下来又可以按 $(D\mathrm{mod}N)+1$，$(2D\mathrm{mod}N)+1$，……的顺序进行 $n$ 轮跳跃。
每一轮 $n$ 个直到所有格子都被遍历到。所以我们只需要确定我们在哪一轮，偏移量 + 轮数即可
轮数：$len=(K-1)/n$。
偏移量：$add=(K-1)\mathrm{mod}n$。
$ans=add\ast D$。

最初容易想到特殊情况， 当$D$ 为 $N$ 的因子时，前 $N/D$ 步肯定没有重复，接下来每一步都是上一轮同样倍数 + $1$
若为 $0,3,6,9\dots \dots$ 下一轮即为 $1,4,7,10\dots \dots$。
随后再引申到非因子的数应该也有一个循环节上进行考虑。

代码

#include <iostream>
#include <algorithm>
using namespace std;
#define ll long long
inline int gcd(int a, int b){
    if(b) while( (a %= b) && (b %= a));
    return a + b;
}
void solve()
{
    ll N, K, D;
    cin >> N >> D >> K; 
    ll g = gcd(N, D);
    ll n = N / g;
    ll len = (K - 1) / n;
    ll add = (K - 1) % n; 
    cout << add * D % N + len << "\n";
    return ;  
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)solve();
    return 0;
}