P2096 松鼠吃果子

#include <bits/stdc++.h>

using namespace std;

int f[10001] = {0};

int main(){
    int n = 0;
    int m = 0;
    int z = 1;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++){
        f[i] = i;
    }
    for(int i = 1; i <= m; i++){
        z = z + i * i * i % 5 + 1;
        if(f[z] != 0){
            if(i == m){
                printf("%d", f[z]);
                break;
            }
            f[z] = 0;
            for(int j = z + 1; j <= n; j++){
                f[j-1] = f[j];
                f[j] = 0;
            }
        }
        else{
            z = i * i * i % 5 + 2;
            if(i == m){
                printf("%d", f[z]);
                break;
            }
            f[z] = 0;
            for(int j = z + 1; j <= n; j++){
                f[j-1] = f[j];
                f[j] = 0;
            }
        }
    }
    return 0;
}