剑指offer-45-圆圈中最后剩下的数字-java

题目及测试

package sword045;
/* 题目：0,1，...，n-1 这n个数字排成一个圆圈，从数字0开始每次从这个圆圈里删除第m个数字。求出这个圆圈里剩下的最后一个数字。

测试样例：

输入： 0，1 , 2, 3, 4

输出： 3
*/

import java.util.List;

public class main {
	
	public static void main(String[] args) {
		int [] testTable = {5,10,15};
		int [] testTable2 = {1,2,3};
		for(int i=0;i<testTable.length;i++){
			test(testTable[i],testTable2[i]);
		}
	}
		 
	private static void test(int ito,int ito2) {
		Solution solution = new Solution();
		int rtn;
		long begin = System.currentTimeMillis();
		System.out.print(ito+"  ");
		System.out.print(ito2);
		System.out.println();
		//开始时打印数组
		
		rtn= solution.lastRemaining(ito,ito2);//执行程序
		long end = System.currentTimeMillis();	
		
		System.out.println("rtn=" );
		System.out.print(rtn);	
		System.out.println();
		System.out.println("耗时：" + (end - begin) + "ms");
		System.out.println("-------------------");
	}

}

解法1（成功）

既然题目中有一个数字圆圈，很自然的想法就是用个数据结构来模拟这个圆圈。在常用的数据结构中，我们很容易的想到环形链表。我们可以创建一个共有n个结点的环形链表，然后每次都从这个链表中删除第m个结点

package sword045;


public class Solution {
	public int lastRemaining(int n, int m) {
		if(n == 0 || n==1) {
			return 0;
		}
		ListNode first = new ListNode(0);
		ListNode now = first;
		for(int i=1;i<n;i++) {
			now.next = new ListNode(i);
			now = now.next;
		}
		now.next = first;
		now = first;
		for(int i=0;i<n-1;i++) {
			for(int j = 1;j<m;j++) {
				now = now.next;
			}
			removeNode(now);
		}
		return now.val;
	}
	
	private void removeNode(ListNode node) {
		ListNode next = node.next;
		node.val = next.val;		
		node.next = next.next;
		next.next = null;
	}
	
	
}

解法2（别人的）

定义函数f(n,m)，表示每次在n个数字0,1,...,n-1中每次删除第m个数字最后剩下的数字。
在n个数字中，第一个被删除的数字是（m%n）-1，我们把这个数字记为K. 在删除掉第一个元素K后，剩下的n-1个数字就是0,1,2,...,k-1,k+1,...,n-1，并且下一次删除从K+1开始计数。那么，在下一次计数的时候其实就相当于在这样一个序列中遍历：K+1，...，n-1，0, 1，... ,  K-1 。这个序列和前一个序列其实是一样的，不一样的是我们把它的顺序修改了一下而已，但是删除元素时遍历顺序是一样的。故经过若干次删除后剩下的数字和前一个序列也应该是一样的。我们把后一个序列每次删除第m个数字最后剩下的数字记为f'(n-1,m)，至于为什么记为f'(n-1,m)你看到后面就懂了。那么现在我们最起码可以确定的是f(n,m)=f'(n-1,m)。

我们再来看分析这个序列：k+1，...，n-1，0，1，...，k-1 。我们将这个序列做一个映射，映射结果是形成一个从0到n-2的序列：
       k+1     ->    0
       k+2     ->    1
              ......
       n-1     ->     n-k-2
       0        ->     n-k-1
       1        ->     n-k
              ......
       k-1      ->     n-2
    f'(n-1,m)     f(n-1,m)
我们定义映射为p，那么p(x) = (x-k-1)%n 。 它表示如果映射前的数字是x，那么映射后的数字是(x-k-1)%n。该映射的逆映射是p-1(x)=(x+k+1)%n。既然要掌握这个方法，就要彻底搞懂，下面跟着我一起证明一遍：

证明：
令y = p(x)，即 y = (x-k-1)%n
则有  y = (x-k-1) +t1n，t1属于整数，且0<= y <n  
< --->   x =  y - t1n + k + 1
<---->   x =  (y+k+1) + t2n ，即 y = (x+k+1) + tn，故p-1(x) =  (x+k+1) %n
证明完毕。

现在，我们发现经过映射之后的n-1个数字是不是和原先的n个数字形式上是一样的？只不过少看一个数字n-1而已。那么，对0，1，...，n-2这n-1个数字，排成一个圆圈，从数字0开始每次删除第m个数，剩下的数字是不是可以表示成f(n-1,m)?! 现在有没有发现我们之前为什么要定义那么序列为 f'(n-1,m)? 这是要建立两次删除之间的联系！就是说原始的n个元素，在删除第一个元素k之后，按理说初始序列已经被打乱了，没有规则了；但是我们通过一个映射关系，让序列重新排列成初始序列的形式。这样只要我们找到这样的映射关系，求出两次操作之间的函数关系（迭代规律）就将问题转化成了递归问题。而递归问题的出口很好确定，当n=1时，序列只有一个元素：0，f(1,m)就是这个0！
既然有了映射关系，下面我们求两次迭代操作之间的关系，即如何由f(n-1,m)求得f(n,m)。

求解：
因为f(n,m) = f'(n-1,m)，且f'(n-1,m) = (f(n-1,m)+k+1)%n，故f(n,m) = (f(n-1,m)+k+1)%n。 又因为 k = (m%n)-1，带入f(n,m) = (f(n-1,m)+k+1)%n，得：f(n,m) = (f(n-1,m)+m)%n。
因此，当n=1时，f(n,m) = 0
当n>1时，f(n,m) = [f(n-1,m)+m]%n

有了这个递推关系，是不是可以写代码了？可以由上而下的用递归，也可以由下而上的用迭代。递归在这里显然不存在子问题重复求解的问题，但是会有大量的堆栈操作，不如直接用迭代的方式。至于迭代方式的源码，上面已经给出了。 int last = 0; 是当n=1时,f(1,m)的值；后面的for循环就是自下而上的求解f(n,m）的值了。
这种思路非常复杂，但是代码尤其简洁，主要的时间都花在了分析和推导公式上了。该方法时间复杂度为O（n），空间复杂度是O（1），无论是时间复杂度和空间复杂度都要好于第一种方法

public static int lastRemaining_2(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;
    }