1045. 快速排序(25) python篇

╰+哭是因爲堅強的太久メ 2022-07-03 12:20 168阅读 0赞

# 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（<= 105）；第2行是空格分隔的N个不同的正整数，每个数不超过109。
    
    **输出格式：**
    
    在第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))