【约瑟夫环】2018-2019 ACM-ICPC, Asia Shenyang Regional Contest - K - Let the Flames Begin

题目链接https://codeforces.com/gym/101955/problem/K

---

题意

给出一个长度为$n$的约瑟夫环，标号为$k$的人被淘汰。问第$m$个人淘汰的初始标号。

$1\le n,m,k\le 1e18$，$\sum min(m,k)\le 2e6$

---

题解

初始标号如果以零开头的话，约瑟夫环有一个公式：$dp[n][m]=(dp[n-1][m-1]+k)\%n$，其中$dp[n][m]$表示以用有$n$个人，第$m$个人淘汰的初始标号。
很容易求得$dp[n-m+1][1]$，然后更新到$dp[n][m]$即可，更新分几种情况

• 如果$m<k$，直接暴力更新
• 如果$m>k$，那么就代表$n$相对于$k$很大，可以算出取模内能加上多少个$k$，然后转化成乘法。
• $k=1$，特判一下

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=2e6+7;
ll t,cs;
ll n,m,k;
int main(){
    scanf("%lld",&t);
    while(t--){
        scanf("%lld%lld%lld",&n,&m,&k);
        if(k==1){
            printf("Case #%lld: %lld\n",++cs,m);
            continue;
        }
        if(m<=k){
            ll ans=(k-1)%(n-m+1);
            for(ll i=2,j=n-m+2;i<=m;i++,j++){
                ans=(ans+k)%j;
            }
            printf("Case #%lld: %lld\n",++cs,ans+1);
        }
        else{
            ll len=n-m+1;
            ll ans=(k-1)%(n-m+1);
            m--;
            while(m>0){
                ll x=(len-ans)/(k-1);
                if(m<=x){
                    ans=(ans+m*k)%(len+x);
                    m=0;
                }
                else{
                    ans=(ans+x*k)%(len+x);
                    ans=(ans+k)%(len+x+1);
                    m-=x+1;
                    len+=x+1;
                }
            }
            printf("Case #%lld: %lld\n",++cs,ans+1);
        }
    }
}