题目大意:
给出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;
}