剑指Offer-46.孩子们的游戏(圆圈中最后剩下的数)(C++/Java)
题目:
分析:
最先想到的就是利用队列,将小朋友序号添加进队列中,然后每由队首的开始报数,出队,再添加进队尾。当报数m-1时,队首的那个序号直接出队,不再进入队尾,继续执行直到队列中剩下一个序号,这个就是最后的答案。
看了下书,有一种利用推导求得答案的方法,就是删除的序号假如为k,实际上k=(m-1)%n,因为下一轮报数开始从k+1这个序号开始,将k+1…n-1,0,1,…k-1映射到0,1,2,…,n-2再利用k=(m-1)%n求得下一轮删除的序号,实际上也就是将最后剩下的那个序号当成0,反推回n个人时,为多少。
程序:
C++
//use queue class Solution { public: int LastRemaining_Solution(int n, int m) { if(n < 1 || m < 0) return -1; queue<int> q; for(int i = 0; i < n; i++) q.push(i); int num = 0; while(num < m){ if(q.size() == 1) return q.front(); if(num == m-1){ q.pop(); num = 0; } else{ auto temp = q.front(); q.pop(); q.push(temp); num++; } } } };
//f(n,m) =(f(n-1,m)+m) %n class Solution { public: int LastRemaining_Solution(int n, int m) { if(m < 0 || n < 1) return -1; if(n == 1) return 0; return (LastRemaining_Solution(n-1, m) + m) % n; } };
Java
//use ArrayList import java.util.*; public class Solution { public static int LastRemaining_Solution(int n, int m) { if(n < 1 || m < 0) return -1; ArrayList<Integer> list = new ArrayList<>(); for(int i = 0; i < n; i++) list.add(i); int index = 0; while(list.size() > 1) { index = (index + m - 1) % list.size(); list.remove(index); } return list.get(0); } }