1045. 快速排序(25) python篇

1045. 快速排序(25)

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CAO, Peng

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

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

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

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

  输入格式：

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

  输出格式：

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

  输入样例：

  5
  1 3 2 4 5
  

  输出样例：

  3
  1 4 5
  

整了一上午，参考了别人的代码，写了两种方案仍然得不到满分，第一种方案对的测试点第二种方案错，第二点种方案对的测试点第一种方案错，弄了半天也不知在哪里不对，希望有人能解答一下。

第一种方案（主元位置固定法）

n=int(input())
m=list(map(int,input().split()))
k=sorted(m)
a=[]
for i in range(n):
    if m[i]==k[i]:
        flag=0
        for j in range(i):
            if m[j]>m[i]:
                flag=1
                break
        if flag==0:
            a.append(str(m[i]))
print(len(a))
if len(a)==0:
    print("")
else:
    print(" ".join(a))

第二种方案（单次遍历排除法）http://www.cnblogs.com/hkevin/p/5510063.html

n=int(input())
m=list(map(int,input().split()))
a=[]
max=0
sum=0
for i in range(n):
    if m[i]>max:
        max = m[i]
        a.append(m[i])
        sum+=1
    else:
        for j in range(len(a)-1,-1,-1):
            if m[i]<a[j]:
                a[j]=0
                sum-=1
            else:
                break
a.sort()
print(sum)
if sum==0:
    print("")
else:
    for i in a:
        if i==0:
            a.remove(i)
    a=list(map(str,a))      
    print(" ".join(a))