HDU 5860 - Death Sequence ( 递推＋约瑟夫环 )

题意

给出编号为1–n的n个人, 每隔k个杀一个人, 询问m次, 问第x个杀掉的是谁

思路

比赛的时候推递推式没推出来, 只好用vector的erase试试, 当然是超时了
比较像约瑟夫环, 但这个题目是不成环的
赛后还是查了查正解, 发现确实是有递推关系的
将编号改成从0开始。如果i%k==0, 那么i第一轮就被杀死。如果i%k!=0，否则在下轮中编号为i-i/k-1

AC代码

参考代码 : hdu 5860 Death Sequence（2016 Multi-University Training Contest 10——递推）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn = 3000000+5;

struct death{
    int pos;
    int d;
    int num;
    //编号为pos的人在d轮第num个被杀
}s[maxn];

bool cmp(struct death a, struct death b){
    if( a.d == b.d ) return a.num < b.num;
    return a.d < b.d;
}

int main()
{
    int T;
    int n, k, q, a;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d",&n, &k, &q);
        s[0].num = 1;
        for(int i = 0; i < n; i++){
            s[i].pos = i+1;
            if( i % k == 0 ){
                s[i].d = 1;
                if( i==0 )continue;
                s[i].num = s[i-k].num+1;
            }
            else{
                s[i].d = s[i-i/k-1].d+1;
                s[i].num = s[i-i/k-1].num;
            }
        }
        sort(s, s+n, cmp);
        while(q--){
            scanf("%d",&a);
            printf("%d\n",s[a-1].pos);
        }

    }
    return 0;
}