剑指offer 6.4 抽象建模能力3 -圆圈中最后剩下的数字

面试题45： 圆圈中最后剩下的数字

题目：n个数字（0,1,…,n-1）形成一个圆圈，从数字0开始，每次从这个圆圈中删除第m个数字（第一个为当前数字本身，第二个为当前数字的下一个数字）。当一个数字删除后，从被删除数字的下一个继续删除第m个数字。求出在这个圆圈中剩下的最后一个数字。

分析：这是一个约瑟夫环问题。

解法一：使用环形链表模拟圆圈。创建一个总共有n个节点的环形链表，每次在这个链表中删除第M个节点。

如果没有特殊要求，可以使用STL中的std::list来模拟一个环形链表。

注意：list本身不是一个环形结构，因此每当迭代器扫描到链表末尾时，要把迭代器移到链表的头部，这样就相当于按照顺序在一个圆圈里遍历了。

int lastRemain(const int n, const 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 current = numbers.begin();
	while(numbers.size() > 1) {
		for(int j = 0; j < m; j ++) {
			current ++;
			if(current == numbers.end()) {
				current = numbers.begin();
			}
		}
		list<int>::iterator next = ++current;
		if(next == numbers.end()) {
			next = numbers.begin();
		}
		--current;
		numbers.erase(current);
		current = next;
	}
	return *(current);
}

每删除一个数字需要m步运算，总共有n个数字，因此总的时间为O(mn)。

还需要一个辅助链表来模拟圆圈，空间复杂度为O(n);

解法二：分析每次被删除的数字的规律，直接计算出圆圈中最后剩下的数字c

首先定义最初的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。我们把这种关系表示为：

                 0                  n=1
f(n,m)={
                 [f(n-1,m)+m]%n     n>1

int LastRemaining_Solution2(int n, unsigned int m)
{
      // invalid input
      if(n <= 0 || m < 0)
            return -1;

      // if there are only one integer in the circle initially,
      // of course the last remaining one is 0
      int lastinteger = 0;

      // find the last remaining one in the circle with n integers
      for (int i = 2; i <= n; i ++) 
            lastinteger = (lastinteger + m) % i;

      return lastinteger;
}