[数据结构]算法设计题--寻找主元素

问题

已知一个整数序列A=(a0,a1,…,an-1)，其中0<=ai<n(0<=i<n)。若存在ap1=ap2=ap3=apm=x其m>n/2，则称x为A的主元素。例如有A=(0,5,5,3,5,7,5,5)，
则5位主元素；又如A=(0,5,5,3,5,1,5,7)，A中没有主元素。假设A中的n个元素保存在一个一维数组中，请设计尽可能高效的算法，找出A的主元素。若存在主元素，
则输出主元素；否则输出-1。要求：

1. 给出算法的基本设计思想
2. 说明设计的算法的时间复杂度和空间复杂度

解答

算法思想

主元素是数组中出现次数超过一半的元素，当数组中存在主元素时，所有非主元素的个数和必然少于一半，如果让主元素与非主元素配对，则最后多出的元素就是主元素。

1. 选定候选主元素，从前往后扫描。
2. 假定主元素key，count计数为1。
3. 遇到下一个整数仍为key，计数加1；否则，计数减1。
4. 计数减到0，下一个整数设置为主元素，计数重新记为1，重复上述步骤。
5. 判断key是否是真正的主元素，如果是返回；否则返回-1。

算法分析

算法时间复杂度O(n)，空间复杂度O(1)。

/*
已知一个整数序列A=(a0,a1,...,an-1)，其中0<=ai<n(0<=i<n)。若存在ap1=ap2=ap3=apm=x其m>n/2，则称x为A的主元素。例如有A=(0,5,5,3,5,7,5,5)，
则5位主元素；又如A=(0,5,5,3,5,1,5,7)，A中没有主元素。假设A中的n个元素保存在一个一维数组中，请设计尽可能高效的算法，找出A的主元素。若存在主元素，
则输出主元素；否则输出-1。要求：
6. 给出算法的基本设计思想
7. 说明设计的算法的时间复杂度和空间复杂度
*/
#include <iostream>
using namespace std;
/*
求数组A中的主元素
*/
int mainElement(int a[], int n) {
	int count = 1; // 计数
	int key = a[0]; // 假定第一个元素为主元素
	for (int i = 1; i < n; ++i) {
		if (key == a[i]) { // 如果是假定主元素，计数+1
			count++;
		} else { // 如果不是主元素，计数-1，如果计数=0，将此数设置为主元素
			if (count > 0) {
				count--;
			} else {
				key = a[i];
				count = 1;
			}
		}
	}
	// 计算主元素的个数
	if (count > 0) {
		count = 0;
		for (int i = 0; i < n; ++i) {
			if (a[i] == key) {
				count++;
			}
		}
	}
	// 超过数组个数的一半则为主元素
	if (count > n / 2) {
		return key;
	} else {
		return -1;
	}
}
int main() {
	int a[100];
	int n;
	cin >> n;
	for (int i = 0; i < n; ++i) {
		cin >> a[i];
	}
	int ans = mainElement(a, n);
	if (ans == -1) {
		cout << "数组A没有主元素" << endl;
	} else {
		cout << "数组A的主元素是" << ans << endl;
	}
}