uvalive4727

题目大意:
给出n,k,n表示的是数字的个数,数字都是从1开始的。
k表示的是每隔几个挑选出一个数字。
输出最后的三个数字。

思路:
约瑟夫环问题。
类似于uvalive3882

代码:

#include <iostream>
using namespace std;
#include <cstring>
#include <stdio.h>

int dp[500005];
int main() {

    int T;
    int n,m;
    scanf("%d",&T);
    while(T--) {
        int n,m,s = 0,ans[3] = {0};
        int pos1,pos2,pos3;
        scanf("%d %d",&n,&m);
        dp[1] = 0;
        for(int i = 2; i <= n; i++) {
            dp[i] = (dp[i - 1] + m)%i;
        }
        ans[0] = dp[n] + 1;
        pos1 = dp[2] = (0 + m) % 2;
        pos2 = dp[2] = (0 + m) % 2 == 1? 0 : 1;
        dp[2] = pos2;
        for(int i = 3; i <= n; i++)
            dp[i] = (dp[i - 1] + m)%i;
        ans[1] = dp[n] + 1;
        bool vis[3] = {0};
        vis[(pos1 + m)%3] = 1;
        vis[(pos2 + m)%3] = 1;
        for(int i = 0; i < 3; i++)
            if(vis[i] == 0)
                pos3 = i;
        dp[3] = pos3;
        for(int i= 4; i <= n; i++)
            dp[i] = (dp[i - 1] + m)%i;
        ans[2] = dp[n] + 1;
        printf("%d %d %d\n",ans[2],ans[1],ans[0]);
    }
    return 0;
}

你可能感兴趣的:(uvalive4727)