剑指offer - 46 -- 孩子们的游戏（圆圈中最后剩下的数） - C++

题目描述

每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0...m-1报数....这样下去....直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢？(注：小朋友的编号是从0到n-1)

 

如果没有小朋友，请返回-1

示例1

输入

5,3

返回值

3

题解

一种方法是用环形链表模拟圆圈的经典解法：

class Solution {
public:
    int LastRemaining_Solution(int n, int m)
    {
        if(n <= 0 || m <= 0)
            return -1;
        ListNode *head = new ListNode(0);
        ListNode *node = head;
        for(int i=1; i<n; i++){    //把n个数都放进链表中
            node->next = new ListNode(i);
            node = node->next;
        }
        node->next = head;  // 链表头尾相连，形成一个环形链表
        
        int k=0;
        while(node->next != node){   //当链表中只剩下一个数字是退出循环
            if(k == m-1){     //当第k个数刚好等于m-1时，删除第k个数，并从k+1又重新从0开始数
                node->next = node->next->next;
                k = 0;
            } else{
                node = node->next;
                k += 1;
            }
        }
        return node->val;
    }
};

或者使用vector来存储所有数字，然后每次删除的是第(start+m-1)%m个数，代码如下：

class Solution {
public:
    int LastRemaining_Solution(int n, int m)
    {
        if(n <= 0 || m <= 0)
            return -1;
        vector<int> circle;
        for(int i=0; i<n; i++)     //把所有数字都放入vector中
            circle.push_back(i);
        
        int start = 0;
        while(n > 1){
            start = (start + m -1) % n;    //下一次开始的地方都是本次删除数字的下一个
            circle.erase(circle.begin() + start);
            n -= 1;
        }
        return circle[0];
    }
};

第二种是分析每次被删除数字的规律并直接计算圆圈中最后剩下的数字。(我没看明白，有没有同学可以给我讲一下)

f[1] = 0
f[2] = (f[1] + m) % 2
f[3] = (f[2] + m) % 3
...
f[n] = (f[n-1] + m) % n
所以代码如下：

class Solution {
public:
    int LastRemaining_Solution(int n, int m)
    {
        if(n <= 0 || m <= 0)
            return -1;
        int index = 0;
        for(int i=2; i<=n; i++){
            index = (index + m)%i;
        }
        return index;
    }
};