2

我在 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

这种方法哪里出错了？

4

5 回答 5

3

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

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
        else:
            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) <= 1:
        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)
于 2014-05-13T02:28:44.523 回答
1

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

if len(lis)<2:
    return lis
于 2013-02-06T15:58:14.033 回答
1

这一行：

return lis

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

return m,c
于 2013-02-06T15:58:24.827 回答
1

代替

return lis

做

return lis, count
于 2013-02-06T15:59:44.373 回答
1

好吧，你在他期望两个的地方返回一个值。

看着

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) 将 < 2 并且它将返回单元素列表，但不会返回计数，这在下面的调用中是预期的。

只需为 c1 返回一个值，例如

return lis, count # do your things with count
于 2013-02-06T16:00:07.277 回答