我正在使用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
我相信我已经找到了关键问题: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]