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

題目

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

如果沒有小朋友，請返回-1

思路

約瑟夫問題的和諧版…每次會輪到誰並不需要傻傻地去模擬m次報數，可以直接計算出來。

index(0)=0
index(t)=(index(t-1)+m-1)%剩餘人數

代碼

import java.util.ArrayList;
public class Solution {
    public int LastRemaining_Solution(int n, int m) {
        if(n<1||m<1)
            return -1;
        ArrayList<Integer> list = new ArrayList<>();
        for(int i=0;i<n;i++)
            list.add(i);
        int index=0;
        while(list.size()>1){
            index = (index+m-1)%list.size();
            list.remove(index);
        }
        return list.get(0);
    }
}