劍指Offer——孩子們的遊戲(圓圈中最後剩下的數)

題目描述

每年六一兒童節,牛客都會準備一些小禮物去看望孤兒院的小朋友,今年亦是如此。HF作爲牛客的資深元老,自然也準備了一些小遊戲。其中,有個遊戲是這樣的:首先,讓小朋友們圍成一個大圈。然後,他隨機指定一個數m,讓編號爲0的小朋友開始報數。每次喊到m-1的那個小朋友要出列唱首歌,然後可以在禮品箱中任意的挑選禮物,並且不再回到圈中,從他的下一個小朋友開始,繼續0…m-1報數…這樣下去…直到剩下最後一個小朋友,可以不用表演,並且拿到牛客名貴的“名偵探柯南”典藏版(名額有限哦!!^_)。請你試着想下,哪個小朋友會得到這份禮品呢？(注：小朋友的編號是從0到n-1)

題解

#include <iostream>
#include <vector>

using namespace std;

int LastRemaining_Solution(int n, int m) {
    if (n < 1 || m < 1)return -1;
    int count = n, step = 0, i = -1;
    vector<int> v(n);
    while (count) {
        i++;
        i = i % n;
        if (v[i] == -1)continue;
        step++;
        if (step == m) {
            v[i] = -1;
            step = 0;
            count--;
        }
    }
    return i;
}

int main() {
    ios::sync_with_stdio(false);
    int n, m;
    cin >> n >> m;
    cout << LastRemaining_Solution(n, m);
    return 0;
}