1、問(wèn)題描述
合并兩個(gè)升序的整數(shù)數(shù)組A和B，形成一個(gè)新的數(shù)組，新數(shù)組也要有序。
2、問(wèn)題實(shí)例
輸入A=[1]，B=[1], 輸出[1，1]，返回合并后的數(shù)組，輸入A=[1,2,3,4], B=[2,3,4,5], 輸出[1，2，2，3，4，4，5，6]， 返回合并所有元素后的數(shù)組。
3、代碼實(shí)現(xiàn)
書(shū)中代碼：
class Solution():
# 參數(shù)A：有序整數(shù)數(shù)組A
# 參數(shù)B：有序整數(shù)數(shù)組B
# 返回：一個(gè)新的有序整數(shù)數(shù)組
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))
輸出結(jié)果為：
在這里插入圖片描述
書(shū)中代碼解讀：
題中A和B數(shù)組均為升序數(shù)組，書(shū)中采用多個(gè)while實(shí)現(xiàn)，第一個(gè)while的意思是，如果A列表中的第一個(gè)數(shù)比B列表中的第一個(gè)數(shù)小，那么，將這個(gè)數(shù)添加到c列表中，接下來(lái)，用A列表的第二個(gè)數(shù)與B中的第一個(gè)數(shù)做比較，如果B列表的第一個(gè)數(shù)比A列表中的第二數(shù)小，那么將該數(shù)添加到c列表中，依次類推。第二個(gè)while和第三個(gè)while是用來(lái)處理A列表和B列表中相同數(shù)的。
3、代碼思考與拓展
1、代碼思考
書(shū)中代碼的前提條件是A和B列表是升序列表，若A和B列表不是升序列表，則上述類不適用，可對(duì)列表A和B進(jìn)行sort()處理，將A和B變?yōu)樯蛄斜?。舉例如下：
A = [1, 6, 3, 4, 1, 9, 2]
B = A.copy()
B.sort()
print(A)
print(B)
輸出為：
在這里插入圖片描述
注意在排序之前，對(duì)A列表進(jìn)行copy，防止后續(xù)處理對(duì)A列表帶來(lái)的影響。
2、代碼拓展
基于此，我提出一個(gè)新要求，給定一個(gè)并沒(méi)按照升序排列的數(shù)字列表，記下此時(shí)的索引，要求該列表按照一定要求升序或者降序之后，還能和其之前未升序或降序前的索引對(duì)應(yīng)。
比如：給定列表A=[2, 5, 4, 9, 1], 則數(shù)字2對(duì)應(yīng)索引為0，數(shù)字5對(duì)應(yīng)索引為1，對(duì)A進(jìn)行升序排列得到列表B=[1, 2, 4, 5, 9], 此時(shí)我們?nèi)阅苤罃?shù)字1對(duì)應(yīng)的先前索引。
解決方法：利用pandas。
import pandas as pd
# 對(duì)于單個(gè)列表
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)
# 對(duì)于列表嵌套列表
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)
輸出結(jié)果為：
在這里插入圖片描述
