1045 快速排序（25 分）

著名的快速排序算法里有一个经典的划分过程：我们通常采用某种方法取一个元素作为主元，通过交换，把比主元小的元素放到它的左边，比主元大的元素放到它的右边。 给定划分后的 N 个互不相同的正整数的排列，请问有多少个元素可能是划分前选取的主元？

例如给定 $N = 5$, 排列是1、3、2、4、5。则：

• 1 的左边没有元素，右边的元素都比它大，所以它可能是主元；
• 尽管 3 的左边元素都比它小，但其右边的 2 比它小，所以它不能是主元；
• 尽管 2 的右边元素都比它大，但其左边的 3 比它大，所以它不能是主元；
• 类似原因，4 和 5 都可能是主元。

因此，有 3 个元素可能是主元。

输入格式：

输入在第 1 行中给出一个正整数 N（≤10​5​​）； 第 2 行是空格分隔的 N 个不同的正整数，每个数不超过 10​9​​。

输出格式：

在第 1 行中输出有可能是主元的元素个数；在第 2 行中按递增顺序输出这些元素，其间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

5
1 3 2 4 5

输出样例：

3
1 4 5

 分析：

这道题很简单，先上代码：

// B45.cpp : Defines the entry point for the console application.
//

#include <iostream>
using namespace std;

int main() 
{
	int n;
	cin >> n;
	int a[10005] = { 0 };
	bool flag = 0;
	int num = 0;
	int b[10005];
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
	}
	for (int i = 0; i < n; i++)
	{
		//第一个元素
		if (i == 0)
		{
			flag = 0;
			for (int after = 1; after < n; after++)
			{
				if (a[i] >= a[after])
				{
					flag = 1;
				}
			}
			if (flag == 0)
			{
				b[num] = a[i];
				num++;
			}
		}
		//中间元素
		else if (i > 0 && i < n)
		{
			flag = 0;
			for (int before = 0; before < i; before++)
			{
				if (a[i] <= a[before])
				{
					flag = 1;
				}
			}
			for (int after = i + 1; after < n; after++)
			{
				if (a[i] >= a[after])
				{
					flag = 1;
				}
			}
			if (flag == 0)
			{
				b[num] = a[i];
				num++;
			}
		}
		//最后一个元素
		else if (i == n)
		{
			flag = 0;
			for (int before = 0; before < i; before++)
			{
				if (a[i] <= a[before])
				{
					flag = 1;
				}
			}
			if (flag == 0)
			{
				b[num] = a[i];
				num++;
			}
		}
	}
	cout << num << endl;
	for (int i = 0; i < num; i++)
	{
		if (i != num - 1)
		{
			cout << b[i] << " ";
		}
		else
		{
			cout << b[i];
		}
	}
	system("pause");
	return 0;
}

结果只通过了一个测试：

这个段错误真的是搞得我一头雾水，后来查了很多的资料，发现：段错误是因为“访问了不该访问的内存”但是我还是找不到到底是哪里出错了。

看了一下别人的写法：

如果能想到，主元的位置与排完序后该元素所在位置相同，那么再满足它是它之前所有元素中最大的一个，就可以断定它是主元。。。还有硬算肯定是会超时的。。。

#include <iostream>
#include <algorithm>
#include <stdio.h>
using namespace std;
 
int main()
{
   int N;
   scanf("%d",&N);
   int num[N],num_s[N];  //一个原始数据，另一个保存已经排好的数据
   for(int i=0;i<N;i++)
   {
       scanf("%d",&num[i]);
       num_s[i]=num[i];
   }
   sort(num_s,num_s+N);
   int count=0,max=0;
   int res[N];
   
   for(int i=0;i<N;i++)
   {
       if(num[i]>max)
            max=num[i];
        if(max==num[i] && num[i]==num_s[i])  //如果当前数是从第一个数到当前数最大的一个，且与排完顺序对应位置的数相同则该数就有可能是主元
            res[count++]=num[i];
   }
    printf("%d\n",count);
    //sort(res,res+count);
    for(int i=0;i<count;i++)
        if(i==0)
             printf("%d",res[i]);
        else
            printf(" %d",res[i]);
    printf("\n");
    return 0;
}