面试题62-圆圈中最后剩下的数字

题目

0,1,···,n-1这n个数字排成一个圆圈，从数字0开始，每次从这个圆圈里删除第m个数字（删除后从下一个数字开始计数）。求出这个圆圈里剩下的最后一个数字。
例如，0、1、2、3、4这5个数字组成一个圆圈，从数字0开始每次删除第3个数字，则删除的前4个数字依次是2、0、4、1，因此最后剩下的数字是3。

解题思路

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

我们可以创建一个共有n个节点的环形链表，然后每次在这个链表中删除第m个节点。
如果面试官没有特殊要求，那么可以用模板库中的std::list来模拟一个环形链表。由于std::list本身并不是一个环形结构，因此每当迭代器扫描到链表末尾的时候，我们要记得把迭代器移到链表的头部，这样就相当于按照顺序在一个圆圈里遍历了。

class Solution
{
public:
	int LastRemaining(unsigned int n, unsigned int m)
	{
		if (n < 1 || m < 1)
			return -1;
		unsigned int i = 0;
		list<int>numbers;
		for (int i = 0; i < n; ++i)
			numbers.push_back(i);
		list<int>::iterator current = numbers.begin();		
		while (numbers.size() >1)
		{
			for (int i = 1; i < m; ++i)
			{
				current++;
				if (current == numbers.end())   //end相当于numbers最后一个元素的下一个位置
					current = numbers.begin();
			}
			list<int>::iterator next = ++current;
			if (next == numbers.end())
				next = numbers.begin();
			--current;
			numbers.erase(current);
			current = next;
		}
		return *current;
	}
};

1. 时间复杂度：O（mn），链表循环插入需要O（n）的时间，执行删除第m个节点需要O（（n-1）*(m-1)），所以时间复杂度是O（mn）
2. 空间复杂度：O（n）

方法二：数学归纳法+递归

题目中的要求可以表述为：给定一个长度为 n 的序列，每次向后数 m 个元素并删除，那么最终留下的是第几个元素？

这个问题很难快速给出答案。但是同时也要看到，这个问题似乎有拆分为较小子问题的潜质：如果我们知道对于一个长度 n - 1 的序列，留下的是第几个元素，那么我们就可以由此计算出长度为 n 的序列的答案。

算法

我们将上述问题建模为函数 f(n, m)，该函数的返回值为最终留下的元素的序号。

首先，长度为 n 的序列会先删除第 m % n 个元素，然后剩下一个长度为 n - 1 的序列。那么，我们可以递归地求解 f(n - 1, m)，就可以知道对于剩下的 n - 1 个元素，最终会留下第几个元素，我们设答案为 x = f(n - 1, m)。

由于我们删除了第 m % n 个元素，将序列的长度变为 n - 1。当我们知道了 f(n - 1, m) 对应的答案 x 之后，我们也就可以知道，长度为 n 的序列最后一个删除的元素，应当是从 m % n 开始数的第 x 个元素。因此有 f(n, m) = (m % n + x) % n = (m + x) % n。

class Solution {
    int f(int n, int m) {
        if (n == 1) {
            return 0;
        }
        int x = f(n - 1, m);
        return (m + x) % n;
    }
public:
    int lastRemaining(int n, int m) {
        return f(n, m);
    }
};

1. 时间复杂度：O(n)，需要求解的函数值有 n 个。
2. 空间复杂度：O(n)，函数的递归深度为 n，需要使用 O(n) 的栈空间。

方法三：数学归纳法+循环

上面的递归可以改写为迭代，避免递归使用栈空间。

class Solution {
public:
    int lastRemaining(int n, int m) {
        int f = 0;
        for (int i = 2; i != n + 1; ++i) {
            f = (m + f) % i;
        }
        return f;
    }
};