UVA1452 Jump

链接

https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=4198

题解

这题就是约瑟夫问题，问你最后三个出局的人的编号
如图：

每一局我都给他重新编号，这一局编号为$k-1$的人会被处决，这个人的下一个人在下一局会变成$0$号，在还剩下$i$个人的一局，编号为$t$的人，下一局的编号为$(t-k)\mod i$，
$f[i]$表示这一句还有$i$人，这局我还没杀人，我要计算的那个人的编号是多少
最后一个出局的人在最后一局的编号为$0$，所以$f[1]=0$
因为$f[i-1]=f[i]-k\ \ (\mod i)$，所以$f[i]=f[i-1]+k\ \ (\mod i)$
令$f[1]=0$就可以倒推回去最后一一个出局的人在第一局时编号为多少
令$f[1]=1$就可以倒推回去倒数第二个出局的人在第一局时编号为多少
令$f[2]=2$就可以倒推回去倒数第三个出局的人在第一句时编号为多少
时间复杂度$O(n)$

其它设想

树状数组/线段树+二分答案？$O(nlog^2n)$
平衡树？$O(nlogn)$

代码

//DP、推导
#include <cstdio>
#include <cstring>
#include <cctype>
#include <cmath>
#define maxn 500010
#define cl(x,y) memset(x,y,sizeof(x))
#define inf 0x3f3f3f3f
using namespace std;
int f[maxn], N, K;
void init()
{
    scanf("%d%d",&N,&K);
}
void work()
{
    int i;
    f[2]=2;
    for(i=3;i<=N;i++)f[i]=(f[i-1]+K)%i;
    printf("%d ",f[N]+1);

    f[1]=1;
    for(i=2;i<=N;i++)f[i]=(f[i-1]+K)%i;
    printf("%d ",f[N]+1);

    f[0]=0;
    for(i=1;i<=N;i++)f[i]=(f[i-1]+K)%i;
    printf("%d\n",f[N]+1);
}
int main()
{
    int T, i;
    scanf("%d",&T);
    while(T--)init(), work();
}