【问题描述】这是一道2013年考研真题,已知一个整数序列A长度为N,其中若存在a,且a的个数大于N/2,则称a为A的主元素
例如:3 5 5 3 5 7 5 5,其主元素为5
又如:3 5 5 3 5 1 5 7,其中没有主元素。
假设元素保存在一个一维数组中,请设计一个尽可能高效的算法,找出主元素。若存在主元素则输出该元素否则输出
要求时间复杂度为O(N),请注意穷举法时间复杂度是O(N^2),排序再遍历查找的时间复杂度是O(N*logN+N)
【输入形式】
一个整数数组,以0作为结束输入
【输出形式】
主元素,若没有则输出-1
【样例输入】
3 5 5 3 5 7 5 5 0
【样例输出】
5
【样例说明】长度为8,共有5个‘5’
先放完整代码:
#include <iostream>
using namespace std;
int major(int *nums, int len)
{
int candidate = -1;int count = 0;for (int i = 0; i < len; i++){
if (count == 0){
candidate = nums[i];}if (nums[i] == candidate){
count++;}else{
count--;}}count = 0;for (int i = 0; i < len; i++){
if (nums[i] == candidate){
count++;}}return count * 2 > len ? candidate : -1;
}int main()
{
int a[100],temp,len=0;cin >> temp;while (temp != 0){
a[len] = temp;len++;cin >> temp;}cout << major(a,len);
}
这种算法的主要原理是,主要元素必须超过总元素的一半,即主要元素出现的次数要大于其他所有元素之和,在遍历中两两抵消,最后要满足count>=1。
由于不一定会出现主要元素,第一次遍历后选出来的可能是主要元素,也可能是其他数据,所以要第二次遍历来计数,如果满足了count*2大于数组长度,此时candidate这个数据才是主要元素。