如何利用python根据给定的元素顺序，对一个数组进行洗牌

问题

给出两个大小相同的整数数组 arr[]和 index[]， 根据给定的 索引 数组重新排列 arr[] 中的元素 。

输入样本

[10, 11, 12]
[1, 0, 2]

输出示例

11, 10, 12
0, 1, 2

操作方法

我们可以从0到 n-1 （其中 n 是数组的大小）进行遍历，并不断将当前元素放在正确的索引上。这将最终给我们提供所需的数组。为此，我们可以将 index[i] 与 *index[index[i]]*交换 ， 将 arr[index[i] 与 arr[i]交换。］

C++编程

#include<bits/stdc++.h>
using namespace std;
void solve(int a[], int n, int index[]){
    for(int i=0;i<n;i++){
        int temp = index[i];
        swap(index[index[i]], index[i]);
        swap(a[temp], a[i]);
    }
}
int main(){
    int a[] = {50, 40, 70, 60, 90};
    int index[] = {3,  0,  4,  1,  2};
    int n =sizeof(a)/sizeof(a[0]);
    solve(a, n, index);
    for(int i=0;i<n;i++) cout<<a[i]<<" ";
    cout<<"\n";
    for(int i=0;i<n;i++) cout<<index[i]<<" ";
}

输出

40 60 90 50 70 
0 1 2 3 4

C语言编程

#include<stdio.h>
void swap(int *x, int *y)
{
    int temp = *x;
    *x = *y;
    *y = temp;
}

void solve(int a[], int n, int index[]){
    for(int i=0;i<n;i++){
        int temp = index[i];
        swap(&index[index[i]], &index[i]);
        swap(&a[temp], &a[i]);
    }
}
int main(){
    int a[] = {50, 40, 70, 60, 90};
    int index[] = {3,  0,  4,  1,  2};
    int n =sizeof(a)/sizeof(a[0]);
    solve(a, n, index);
    for(int i=0;i<n;i++) printf("%d ", a[i]);
    printf("\n");
    for(int i=0;i<n;i++) printf("%d ", index[i]);
}

输出

40 60 90 50 70 
0 1 2 3 4

Python编程

def solve(a, index, n):
    for i in range(0, n):
        temp = index[i]
        t = index[index[i]]
        index[index[i]] = index[i]
        index[i] = t
        t = a[temp]
        a[temp] = a[i]
        a[i] = t


a = [50, 40, 70, 60, 90]
index = [3,  0,  4,  1,  2]
n = len(a)
solve(a, index, n)
print(a)
print(index)

输出

[40, 60, 90, 50, 70] 
[0, 1, 2, 3, 4]