【发布时间】:2020-04-19 16:18:02
【问题描述】:
给定一个输入数组:
[49, 86, 78]
我的代码应该返回:
[[49,1],[78,2],[86,1]]
也就是说,它应该按照每个子数组中的第一个数字对数组进行排序,第二个将作为索引供以后使用。我决定使用归并排序进行排序。这是我的代码(我知道它很难看,但请运行一次):
def merge_sort(array):
print("array>>>", array)
if len(array[:,0]) == 1:
print('going back')
return
arrL = array[0:int(len(array)/2),:]
arrR = array[int(len(array)/2):,:]
print('left>>', arrL, 'right>>', arrR)
print("*****************")
print('into left')
merge_sort(arrL)
print("*****************")
print('into right')
merge_sort(arrR)
print("*****************")
print('into MERGE')
merge(array, arrL, arrR)
print('going back')
def merge(A, arrL, arrR):
print("array ", A, "left", arrL, "right", arrR)
i = 0
while len(arrL[:,0]) is not 0 and len(arrR[:,0]) is not 0:
print('ARRAYLEFT>>', arrL, 'ARRAYRIGHT>', arrR)
if arrR[0,0] < arrL[0,0]:
print(arrR[0,0], 'is less than', arrL[0,0])
A[i] = arrR[0]
arrR = arrR[1:,:]
print('A>>', A, 'arrR>>', arrR, 'ARRAYLEFT>>', arrL)
else:
print(arrL[0,0], 'is less than', arrR[0,0])
A[i] = arrL[0]
arrL = arrL[1:,:]
print('A>>', A, 'arrL>>', arrL)
i += 1
print('ARRAYLEFT>>', arrL, 'ARRAYRIGHT>>', arrR)
if len(arrL[:,0]) == 0 :
print('left exhaused')
print('A>>', A, 'arrR>>', arrR)
A[i:,:] = arrR
print('A>>', A, 'arrR>>', arrR)
elif len(arrR[:,0]) == 0:
print('right exhaused')
print('A>>', A, 'arrL>>', arrL)
A[i:,:] = arrL
print('A>>', A, 'arrL>>', arrL)
print('after merge', A)
return
import numpy as np
np.random.seed(1)
input_array = [np.random.choice(np.arange(100)) for _ in range(5)]
print(input_array)
输出:
[37, 12, 72, 9, 75]
那么,
new_array = np.array(list( map(list, zip(input_array, [x for x in range(len(input_array)) ])) ))
print(new_array)
输出:
array([[37, 0],
[12, 1],
[72, 2],
[ 9, 3],
[75, 4]])
最后,
merge_sort(new_array)
给出以下内容(我只添加了感兴趣的部分)。 请注意,一旦我们转到into MERGE,一旦 arrR 变为 [],arrL 的值就会变为 arrR 的值,代码中用 <<<=== 符号标记:
array>>> [[37 0]
[12 1]
[72 2]
[ 9 3]
[75 4]]
left>> [[37 0]
[12 1]] right>> [[72 2]
[ 9 3]
[75 4]]
*****************
into left
array>>> [[37 0]
[12 1]]
left>> [[37 0]] right>> [[12 1]]
*****************
into left
array>>> [[37 0]]
going back
*****************
into right
array>>> [[12 1]]
going back
*****************
into MERGE
array [[37 0]
[12 1]] left [[37 0]] right [[12 1]]
ARRAYLEFT>> [[37 0]] ARRAYRIGHT>> [[12 1]] <<<=== # arrL = [[37 0]]
12 is less than 37
A>> [[12 1]
[12 1]] arrR>> [] ARRAYLEFT>> [[12 1]]
ARRAYLEFT>> [[12 1]] ARRAYRIGHT>> [] <<<=== # arrL = [[12 1]]
right exhaused
A>> [[12 1]
[12 1]] arrL>> [[12 1]]
A>> [[12 1]
[12 1]] arrL>> [[12 1]]
after merge [[12 1]
[12 1]]
going back
*****************
## .... and so one....
每当代码进入into MERGE 和arrR = [] 时,似乎都会发生此错误,因此在此之后
print(new_array)
输出:
array([[ 9, 3],
[ 9, 3],
[ 9, 3],
[ 9, 3],
[75, 4]])
这显然是错误的答案。请帮助我理解我在哪里犯了错误......我已经盯着代码和输出看了好 20 分钟,但似乎找不到它。任何建议/批评都将受到高度赞赏。
【问题讨论】:
-
抱歉添加了这么多打印语句...我试图调试代码但无济于事
-
恕我直言,您使此操作过于复杂,对单行解决方案感兴趣(不导入库)?
标签: python recursion mergesort