提问者:小点点

使用中位数作为支点工作的快速排序(但显示comaprisons的计数不正确)


我正在使用python中的快速排序对数字列表进行排序。这必须使用中位数作为支点来完成。我如何找到中位数?它是第二大编号。(在原始列表的第一个、中间和最后一个元素的3个数字列表中)

如果列表长度为偶数(2j),则中间的no将是第j个元素。

如果列表长度为奇数(例如5),则中间的编号将是第3个元素

我的代码工作成功。但是,所做的比较总数不正确(使用变量tot_comparisons找到)

我在干什么?

1.找到支点(即第一个、中间和最后一个元素的中位数)

我不需要代码。我在下面发布了我的帖子,它正在成功工作(除了总数。比较部分)。我只需要知道我代码中的错误

def swap(A,i,k):
    temp=A[i]
    print "temp is "
    print temp
    A[i]=A[k]
    A[k]=temp


def find_median(input_list,start,end):
    #print 'input list is'
    #print input_list

    temp_list1=[]
    temp_list1.append(input_list[start])    
    if len(input_list) % 2 ==0:
        middle_element_index=(len(input_list) / 2)-1
        middle_element=input_list[middle_element_index]
        temp_list1.append(middle_element)
    else:
        middle_element_index=len(input_list)/2
        middle_element=input_list[middle_element_index]
        temp_list1.append(middle_element)    
    temp_list1.append(input_list[end])
    temp_list1.sort()
    #print temp_list1
    if len(temp_list1)==3:
        print temp_list1[1]    
        return temp_list1[1]
    elif len(temp_list1)==2:
        print temp_list1[1]    
        return temp_list1[1]
    else:
        print temp_list1[0]    
        return temp_list1[0] 


def partition(A,start,end,median_index):

    swap(A,start,median_index)
    pivot=A[start]
    pivot_index=start + 1
    for i in range(start+1,end+1):
        if A[i] < pivot:
            swap(A,i,pivot_index)
            pivot_index+=1
    swap(A,pivot_index-1,start)
    return pivot_index-1

def quicksort(A,start,end):
    global tot_comparisons
    if start<end:
        median=find_median(A, start, end)
        median_index=A.index(median)

        pivot_index=partition(A,start,end,median_index)
        tot_comparisons+=end-start
        print "pivot_index"
        print pivot_index
        print "ENDS"


        quicksort(A, start,pivot_index-1)

        #tot_comparisons+=end-pivot_index
        #quicksort(A, pivot_index, end)

        quicksort(A, pivot_index+1, end)

#A=[45,21,23,4,65]

#A=[21,23,19,22,1,3,7,88,110]
#A=[1,22,3,4,66,7]

#A=[1, 3, 7, 19, 21, 22, 23, 88, 110]
#A=[7,2,1,6,8,5,3,4]

temp_list=[]
f=open('temp_list.txt','r')
for line in f:
    temp_list.append(int(line.strip()))
f.close()
print 'list is '
#print temp_list
print 'list ends'

tot_comparisons=0
#quicksort(A, 0, 7)
quicksort(temp_list, 0, 9999)
#quicksort(temp_list, 0, len(temp_list))
print 'hhh'
print temp_list
print tot_comparisons
#print A

共1个答案

匿名用户

我相信我已经找到了关键问题:find_median抓取给定列表的左右元素,然后添加原始列表的中间元素。因此,如果你想对列表位置进行5:7排序,你会抓取元素5、7和3。最后一个应该是元素6。这可能会迫使你完成额外的排序工作。

中间元素有index

(开始结束)/2

(整数除法;看起来您使用的是Python2。?)。对于奇数和偶数长度,您不需要单独的情况。位置取决于给定的列表,而不是原始长度len(input_list)。

只需列出三个需要的元素,对其进行排序,然后返回中间的元素。由于只有一个临时列表,这显然是一个列表,让我们缩短该名称。

temp = [
    input_list[start],
    input_list[end],
    input_list[(start+end) / 2]
]
temp.sort()
return temp[1]

您可以将其简化为一个命令函数。我将把输入列表重命名为简单的in以帮助易读性。

return sorted( [ in[start], in[end], in[(start+end) / 2] ] )[1]