如果一个数组A[1..n]中超过半数的元素都相同时,该数组被称为含有主元素。
算法思想:
利用快速排序的思想,如果这个数组存在主元素,则它一定为排序后的中位数。但问题是,我们要设计O(n)算法,我们知道,排序的最优时间复杂度是O(nlogn),所以我们需要借助其他的方法来完成这个问题。
我们可以想到快速排序,当我们调整基准元素时,如果基准元素到达n/2,则说明左边都比它小,右边都比它大,这个数肯定就是中位数。我们只需要再遍历一次数组,看看数组中和这个中位数相等的数的个数有几个就可以了。如果大于n/2,则含有主元素,反之没有。
时间复杂度为O(nlogn + n)
代码如下:
#include<iostream>
using namespace std;
int n, mid;
int Partition(int arr[], int low, int high)
{
int i, j, pivot;
i = low, j = high, pivot = arr[low];
while(i < j)
{
while(i < j && arr[j] >= pivot) --j;
if(i < j)
swap(arr[i++], arr[j]);
while(i < j && arr[i] <= pivot) ++i;
if(i < j)
swap(arr[i], arr[j--]);
}
return j; //返回基准元素位置
}
void Quick_Sort(int arr[], int low, int high)
{
int pivotpos;
if(low <= high)
{
pivotpos = Partition(arr, low, high);
if(pivotpos > n / 2)
Quick_Sort(arr, low, pivotpos - 1); //基准左递归排序
else if(pivotpos < n / 2)
Quick_Sort(arr, pivotpos + 1, high); //基准右递归排序
else
mid = arr[pivotpos];
}
}
int main()
{
int count = 0;
printf("输入数组元素的个数:");
scanf("%d", &n);
int *a = new int[n];
for(int i = 0; i < n; ++i)
scanf("%d", &a[i]);
Quick_Sort(a, 0, n - 1);
for(int i = 0; i < n; ++i)
if(a[i] == mid)
count++;
printf(count > n / 2 ? "Yes\n" : "No\n");
delete a;
return 0;
}
另一种递归:
#include<iostream>
using namespace std;
int n;
void QuickSort(int arr[], int low, int high)
{
if(low >= high)
return ;
int i, j, pivot;
i = low, j = high, pivot = arr[low]; //每次取中间元素作为基准
while(i < j)
{
while(i < j && arr[j] >= pivot) --j;
arr[i] = arr[j];
while(i < j && arr[i] <= pivot) ++i;
arr[j] = arr[i];
}
arr[i] = pivot;
if(i > n / 2)
QuickSort(arr, low, i - 1); //递归左边
else if(i < n / 2)
QuickSort(arr, j + 1, high); //递归右边
else
return ;
}
int main()
{
int temp, count = 0;
printf("输入数组元素的个数:");
scanf("%d", &n);
int *a = new int[n];
for(int i = 0; i < n; ++i)
scanf("%d", &a[i]);
QuickSort(a, 0, n - 1);
temp = a[n / 2];
for(int i = 0; i < n; ++i)
if(a[i] == temp)
count++;
printf(count > n / 2 ? "Yes,含有主元素\n" : "No,不含有主元素\n");
delete a;
return 0;
}