剑指 Offer（第2版）面试题 62：圆圈中最后剩下的数字

剑指 Offer（第2版）面试题 62：圆圈中最后剩下的数字

题目来源：82. 圆圈中最后剩下的数字

解法1：环形链表

我们可以创建一个有 n 个节点的环形链表，然后每次在这个链表中删除第 m 个节点，直至链表只剩 1 个节点。

STL 的 list 就是环形链表，list.end() 不表示数据，其 next 指针指向 list.begin()。

每当迭代器扫描到链表末尾时，我们需要将其移到链表的头部。

代码：

#include <list>
class Solution
{
public:
	int lastRemaining(int n, int m)
	{
		if (n < 1 || m < 1)
			return -1;
		list<int> numbers;
		for (int i = 0; i < n; i++)
			numbers.push_back(i);
		list<int>::iterator cur = numbers.begin();
		while (numbers.size() > 1)
		{
			for (int i = 1; i < m; i++)
			{
				cur++;
				if (cur == numbers.end())
					cur = numbers.begin();
			}
			list<int>::iterator next = ++cur;
			if (next == numbers.end())
				next = numbers.begin();
			cur--;
			numbers.erase(cur);
			cur = next;
		}
		return *cur;
	}
};

复杂度分析：

时间复杂度：O(n*m)，每删除一个数字需要 m 步运算，共有 n 个数字，因此总的时间复杂度是 O(n*m)。

空间复杂度：O(n)，需要建立一个有 n 个节点的链表。

解法2：数学

首先定义最初的n个数字（0,1,…,n-1）中最后剩下的数字是关于 n 和 m 的方程为f(n, m)。

在这 n 个数字中，第一个被删除的数字是 (m-1)%n，为简单起见记为 k。那么删除 k 之后的剩下 n-1 的数字为0,1,…,k-1,k+1,…,n-1，并且下一个开始计数的数字是 k+1。相当于在剩下的序列中，k+1 排到最前面，从而形成序列 k+1,…,n-1,0,…k-1。该序列最后剩下的数字也应该是关于 n 和 m 的函数。由于这个序列的规律和前面最初的序列不一样（最初的序列是从0开始的连续序列），因此该函数不同于前面函数，记为 f’(n-1,m)。最初序列最后剩下的数字 f(n,m) 一定是剩下序列的最后剩下数字 f’(n-1,m)，所以 f(n,m)=f’(n-1,m)。

接下来我们把剩下的的这 n-1 个数字的序列 k+1,…,n-1,0,…k-1 作一个映射，映射的结果是形成一个从 0 到 n-2 的序列：

k+1    ->    0
k+2    ->    1
…
n-1    ->    n-k-2
0      ->    n-k-1
…
k-1    ->   n-2

把映射定义为 p，则 p(x)= (x-k-1)%n，即如果映射前的数字是 x，则映射后的数字是 (x-k-1)%n。对应的逆映射是p^-1(x)=(x+k+1)%n。

由于映射之后的序列和最初的序列有同样的形式，都是从 0 开始的连续序列，因此仍然可以用函数 f 来表示，记为 f(n-1,m)。根据我们的映射规则，映射之前的序列最后剩下的数字 f’(n-1,m)= p^-1[f(n-1,m)]=[f(n-1,m)+k+1]%n，把 k=(m-1)%n 代入得到 f(n,m)=f’(n-1,m)=[f(n-1,m)+m]%n。

经过上面复杂的分析，我们终于找到一个递归的公式。要得到 n 个数字的序列的最后剩下的数字，只需要得到 n-1 个数字的序列的最后剩下的数字，并可以依此类推。当 n=1 时，也就是序列中开始只有一个数字 0，那么很显然最后剩下的数字就是 0。因此有递推公式：

1. 当 n=1 时，f(n, m) = 0。
2. 当 n>1 时，f(n, m) = [f(n-1, m) +m] % n。

循环代码：

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

递归代码：

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

复杂度分析：

时间复杂度：O(n)。

空间复杂度：O(1)。