2、合并排序数组

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列表在第一次比完大小之后剩下的部分。（剩下的部分是有序的，且都比C数组中的大）

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)

输出结果为：