Question

我已经在Python中创建了一个合并排序程序，它运行得很好但是我已经修改它来计算所涉及的反转次数，现在它给了我一个错误：

这是我的代码：

def merge_list(left,right,c):
    result=[]
    i,j=0,0
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            result.append(left[i])
            print "Left result",result
            i=i+1
        elif left[i] > right[j]:
            result.append(right[j])
            print "Right result",result
            j=j+1
        if right[j] < left[i]  and i<j:
            c=c+1
    result=result+left[i:]
    result=result+right[j:]
    print "Inversions: ",c
    return result,c


def sort_and_count(lis,count):

    if len(lis)<2:
        return lis
    middle=len(lis) / 2
    left,c1=sort_and_count(lis[:middle],count)
    print "left",left
    right,c2=sort_and_count(lis[middle:],count)
    print "right",right
    m,c=merge_list(left,right,count)
    c=c+c1+c2
    return m,c


if __name__=="__main__":

    print "Enter 6 elements: "
    i=0;lis=[];merge_lis=[];inv=0
    while i<=5:
        x=int(raw_input())
        lis.append(x)
        i=i+1
    count=0
    merge_lis,inv=sort_and_count(lis,count)
    print "Sorted list is: ",merge_lis,inv

我的追溯：

Traceback (most recent call last):
  File "Sort_and_count.py", line 53, in <module> 
    merge_lis,inv=sort_and_count(lis,count) 
  File "Sort_and_count.py", line 31, in sort_and_count 
    left,c1=sort_and_count(lis[:middle],count) 
  File "Sort_and_count.py", line 31, in sort_and_count
    left,c1=sort_and_count(lis[:middle],count)
ValueError: need more than 1 value to unpack

这种方法我哪里错了？

Answer 1

实际上，您已经实现了错误的算法来计算反转次数。

1）在函数merge_list而不是：

elif left[i] > right[j]:
    result.append(right[j])
    print "Right result",result
    j=j+1
if right[j] < left[i]  and i<j:
    c=c+1

你应该使用这段代码：

elif right[j] < left[i]:
    result.append(right[j])
    j += 1
    inv_count += (len(left)-i)

2）函数merge_list不需要变量c作为输入。

3）函数sort_and_count不需要变量count作为输入。

试试这段代码：

def merge_list(left,right):
    result = list()
    i,j = 0,0
    inv_count = 0
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            result.append(left[i])
            i += 1
        elif right[j] < left[i]:
            result.append(right[j])
            j += 1
            inv_count += (len(left)-i)
    result += left[i:]
    result += right[j:]
    return result,inv_count


def sort_and_count(array):
    if len(array) < 2:
        return array, 0
    middle = len(array) / 2
    left,inv_left = sort_and_count(array[:middle])
    right,inv_right = sort_and_count(array[middle:])
    merged, count = merge_list(left,right)
    count += (inv_left + inv_right)
    return merged, count





if __name__=="__main__":
    array = [2,3,1,4,5]
    merge_array,inversions = sort_and_count(array)
    print 'Start with array: %s;\nSorted array:     %s;\nInversions: %s.'%(array,merge_array,inversions)

Answer 2

错误消息告诉您sort_and_count仅返回单个值。函数中只有两个返回值，因此罪魁祸首就是这个：

if len(lis)<2:
    return lis

Answer 3

这一行：

return lis

这是一个问题，因为你期望sort_and_count返回一个包含两个值的元组，所以当它只返回一个值时，你会遇到像left,c1=sort_and_count(lis[:middle],count)这样的行解包的问题。该行应返回两个值，如该方法的最后一行：

return m,c

Answer 4

而不是

return lis

DO

return lis, count

Answer 5

嗯，你正在返回一个值，他预计会有两个值。

看看

def sort_and_count(lis,count):
    if len(lis) < 2:
        return lis
    middle = len(lis) / 2
   left, c1 = sort_and_count(lis[:middle],count)
   # etc

如果你调用sort_and_count（[1]，count），那么len（lis）将是＆lt; 2，它将返回单元素列表，但不会返回计数，这在下面的调用中是预期的。

只需返回c1的值，如

return lis, count # do your things with count

使用合并排序计算反转

5 个答案: