对角矩阵按升序排列并对另一同维矩阵执行相同操作的实现方法

你完全不需要纠结归并排序是否适配对角线元素的排序需求，换个思路就能非常简洁地解决问题，不需要在排序过程中实时跟踪、同步交换操作。

核心思路 #

通过索引排序一步完成双矩阵的同步调整：先得到对角线元素升序排列对应的原下标序列，直接用该序列重构对角矩阵、重排另一矩阵的列即可，比实时交换的方案效率更高。

具体实现（Python示例） #

假设你有n阶对角矩阵diag_mat、同维度的待同步列交换矩阵other_mat：

步骤1：生成排序索引 #

import numpy as np
# 提取对角矩阵的对角线元素
diag_vals = np.diag(diag_mat)
# 获取使对角线元素升序排列的原下标序列
sorted_indices = np.argsort(diag_vals)

步骤2：生成排序后的对角矩阵 #

sorted_diag_mat = np.diag(diag_vals[sorted_indices])

步骤3：同步重排另一矩阵的列 #

# 按排序后的下标序列取原矩阵的列，等价于同步所有交换操作
synchronized_other_mat = other_mat[:, sorted_indices]

原理解释 #

以你提到的3阶矩阵为例：假设原对角矩阵的对角线元素为[5,2,7]，调用argsort得到的下标序列为[1,0,2]，该序列表示：排序后第0位的元素来自原下标1的位置、第1位的元素来自原下标0的位置、第2位元素位置不变。直接按该序列提取另一矩阵的列，就等价于完成了你举例的交换第1、2列的操作，不需要记录任何中间交换步骤。

如果你不用numpy，用原生Python实现逻辑也完全一致：把(对角线元素值, 对应下标)打包成元组列表，按元素值升序排序后提取下标序列，再分别生成新的对角矩阵、重排另一矩阵的列即可。

可选：基于交换操作的实现 #

如果你确实需要通过跟踪排序过程的交换步骤实现（比如教学演示需求），直接选用支持原地交换的排序算法即可（比如冒泡排序、选择排序、快速排序）：排序时比较两个下标对应的对角线元素值，每次需要交换两个下标的位置时，同步交换另一矩阵的对应两列即可，完全不需要使用归并排序这类需要额外空间、交换操作不直观的算法。

内容的提问来源于stack exchange，提问作者Shaurya Goyal