二、合并排列數組

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)

輸出結果為:

在這里插入圖片描述
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容