关于水王的题目——找到出现次数大于/等于数组长度一半的数
一个长度为n的数组,里面有一些数出现的次数比较多。请找到出现次数大于数组长度一半的数。(水王1)
一个长度为n的数组,里面有一些数出现的次数比较多。请找到出现次数等于数组长度一半的数。(水王2)
不管是水王1,还是水王2,解法的时间复杂度都是O(n)。
对于水王1,我们用两个变量来解决:candidate and count,前者记录水王的候选者,后者次数他的绝对次数!注意,这里是绝对次数~下面我们来说说具体做法吧:
代码如下:
#include<iostream> using namespace std; int findShuiWang(int *a, int num) { int candidate=a[0]; int count=1; for(int i=1; i<num; i++) { if(candidate==a[i]) count++; else { count--; if(count==0) { candidate=a[i+1]; i++; count=1; } } } return candidate; } int main() { int a[6]={2, 1, 2, 2, 1, 2}; int num=6; int shuiWang; shuiWang=findShuiWang(a, num); cout<<shuiWang<<endl; return 0; }
对于水王2,其出现的次数恰好等于数组长度的一半,这个时候我们可以确定两件事情:
1)水王的“绝对次数”==0;
2)数组的长度一定是个偶数。
同样我们的解法答题按照上面水王1的思路来,一直到最后candidate的count==0,此时要么这个candidate就是水王2,要么就是数组的最后一个数,因为是他才让candidate的count变成0的,到了二选一的节奏了~
此时,我们巧妙的再引入一个变量help,记录数组最后一个数出现的次数,如果等于n/2(注意n一定是偶数),那么我们就认为最后一个数是水王,否则就是candidate是水王~
程序如下:
#include<iostream> using namespace std; int findShuiWang(int *a, int num) { int candidate=a[0]; int count=1; int lastOne=a[num-1]; int helpCount=lastOne==a[0] ? 1 : 0; for(int i=1; i<num-1; i++) //a trick to avoid overflow { if(candidate==a[i]) count++; else { count--; if(count==0) { candidate=a[i+1]; i++; count=1; } } if(a[i]==lastOne) helpCount++; } return helpCount==num/2-1 ? lastOne : candidate; //the last one must be lastOne! } int main() { int a[6]={2, 1, 2, 2, 1, 0}; int num=6; int shuiWang; shuiWang=findShuiWang(a, num); cout<<shuiWang<<endl; return 0; }
原文地址:http://blog.csdn.net/puqutogether/article/details/41845895