合并排序数组
合并排序数组
目录
1、问题描述
2、问题实例
3、代码实现
3、代码思考与拓展
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)
输出结果为：
