根据下标对数组进行排序

追梦少年ML

于 2023-04-25 00:34:37 发布

阅读量396

点赞数

文章标签： python 算法数据结构

本文链接：https://blog.csdn.net/m_l_fighting/article/details/130355449

版权

在不改变数组的情况下，使用下标数组对其进行排序

C++实现

#include<bits/stdc++.h>
using namespace std;
int main(){
    vector<int> nums{1,5,6,9,7,2,3};
    int n = nums.size();
    vector<int> indices(n);
    // 为indices赋值，从0开始，每个元素增加1
    iota(indices.begin(), indices.end(), 0);
    // 在不修改nums数组的情况下，使用下标数组对nums进行排序
    sort(indices.begin(), indices.end(), [&](int x, int y){
        return nums[x] < nums[y];
    });
    for(int i : indices) cout<<i<<" ";
    cout<<endl;
    for(int i : indices) cout<<nums[i]<<" ";
    cout<<endl;
    return 0;
}

Python实现

nums = [1,5,6,9,7,2,3]
indices = list(range(0, len(nums)))
indices.sort(key=lambda x : nums[x], reverse=False)
res = [nums[i] for i in indices]

print("排序后的下标数组：", indices)
print("排序后的数组：", res)