1、問題描述
合并兩個升序的整數數組A和B,形成一個新的數組,新數組也要有序。
2、問題實例
輸入A=[1],B=[1], 輸出[1,1],返回合并后的數組,輸入A=[1,2,3,4], B=[2,3,4,5], 輸出[1,2,2,3,4,4,5,6], 返回合并所有元素后的數組。
3、代碼實現
書中代碼:
class Solution():
# 參數A:有序整數數組A
# 參數B:有序整數數組B
# 返回:一個新的有序整數數組
def Sort1(self, A, B):
i, j = 0, 0
c = []
while i < len(A) and j < len(B):
if A[i] < B[j]:
c.append(A[i])
i += 1
else:
c.append(B[j])
j += 1
while i < len(A):
c.append(A[i])
i += 1
while j < len(B):
c.append(B[j])
j += 1
return c
if __name__ == '__main__' :
A = [1, 4]
B = [1, 2, 4]
D = [1, 2, 3, 4]
E = [2, 4, 6, 8]
solution = Solution()
print('輸入:', A, B)
print('輸出:', solution.Sort1(A, B))
print('輸入:', D, E)
print('輸出:', solution.Sort1(D, E))
輸出結果為:
在這里插入圖片描述
書中代碼解讀:
題中A和B數組均為升序數組,書中采用多個while實現,第一個while的意思是,如果A列表中的第一個數比B列表中的第一個數小,那么,將這個數添加到c列表中,接下來,用A列表的第二個數與B中的第一個數做比較,如果B列表的第一個數比A列表中的第二數小,那么將該數添加到c列表中,依次類推。第二個while和第三個while是用來處理A列表和B列表中相同數的。
3、代碼思考與拓展
1、代碼思考
書中代碼的前提條件是A和B列表是升序列表,若A和B列表不是升序列表,則上述類不適用,可對列表A和B進行sort()處理,將A和B變為升序列表。舉例如下:
A = [1, 6, 3, 4, 1, 9, 2]
B = A.copy()
B.sort()
print(A)
print(B)
輸出為:
在這里插入圖片描述
注意在排序之前,對A列表進行copy,防止后續處理對A列表帶來的影響。
2、代碼拓展
基于此,我提出一個新要求,給定一個并沒按照升序排列的數字列表,記下此時的索引,要求該列表按照一定要求升序或者降序之后,還能和其之前未升序或降序前的索引對應。
比如:給定列表A=[2, 5, 4, 9, 1], 則數字2對應索引為0,數字5對應索引為1,對A進行升序排列得到列表B=[1, 2, 4, 5, 9], 此時我們仍能知道數字1對應的先前索引。
解決方法:利用pandas。
import pandas as pd
# 對于單個列表
A = [2, 5, 4, 9, 1]
C = {'A': A}
df = pd.DataFrame(C)
result = df.sort_values(by='A', axis=0, ascending=True)
print(result)
# 對于列表嵌套列表
a = [[8, 2, 5, 4], [10, 6, 9, 8]]
data = pd.DataFrame(a)
# print(data)
# print(data.T)
data1 = data.T
data1 = data1.sort_values(by=0, axis=0, ascending=True)
print(data1)
輸出結果為:
在這里插入圖片描述