首页 文章资讯内容详情

使用 C++ 重新排列一个数组,以便 arr[i] 变为 arr[arr[i]] 并且 O(1) 额外空间

2026-06-02 1 花语

我们得到一个正整数类型数组,比方说,任何给定大小的arr[],这样数组中的元素的值应该大于0但小于数组的大小。任务是以这样的方式重新排列数组,使得arr[i]仅在给定的O(1)空间内变为arr[arr[i]]并打印最终结果。

让我们看看这个的各种输入输出场景-

输入 -intarr[]={032154}

输出 -排列前的数组:032154重新排列数组,使arr[i]变为arr[arr[i]]并具有O(1)额外空间:012345

解释 -我们得到一个大小为6的整数数组,数组中的所有元素都小于6。现在,我们将重新排列数组,即arr[arr[0]为0,arr[arr[1]]为1,arr[arr[2]]为2,arr[arr[3]]为3,arr[arr[4]]为4,arr[arr[5]]为5。因此,重排后的最终数组为012345。

输入 -intarr[]={1,0}

输出 -排列前的数组:10重新排列数组,使arr[i]变为arr[arr[i]],O(1)额外空间为:01

解释-我们得到一个大小为2的整数数组,数组中的所有元素都小于2。现在,我们将重新排列数组,即arr[arr[0]为1,arr[arr[1]]为0。因此,重排后的最终数组为01。

输入 -intarr[]={1,0,2,3}

输出 -排列前的数组:1023重新排列数组,使arr[i]变为arr[arr[i]],O(1)额外空间为:0123

解释-我们给定一个大小为4的整数数组,数组中的所有元素都小于4。现在,我们将重新排列数组,即arr[arr[0]为0,arr[arr[1]]为1,arr[arr[2]]为2,arr[arr[3]]为3。因此,重排后的最终数组为0123。

下面程序中使用的方法如下

输入整数类型元素的数组并计算数组的大小

排列前打印数组并调用函数Rearrangement(arr,size)

函数内部Rearrangement(arr,size)

从i到0开始循环FOR,直到i小于size。在循环内,将temp设置为arr[arr[i]]%size和arr[i]+=temp*size。

从i到0开始循环FOR,直到i小于size。在循环内,设置arr[i]=arr[i]/size

打印结果。

示例

#include <bits/stdc++.h> using namespace std; void Rearrangement(int arr[], int size){ for(int i=0; i < size; i++){ int temp = arr[arr[i]] % size; arr[i] += temp * size; } for(int i = 0; i < size; i++){ arr[i] = arr[i] / size; } } int main(){ //输入一个数组 int arr[] = {0, 3, 2, 1, 5, 4}; int size = sizeof(arr) / sizeof(arr[0]); //打印原始数组 cout<<"排列前的数组: "; for (int i = 0; i < size; i++){ cout << arr[i] << " "; } //调用函数重新排列数组 Rearrangement(arr, size); //重新排列值后打印数组 cout<<"\nRearrangement of an array so that arr[i] becomes arr[arr[i]] with O(1) extra space is: "; for(int i = 0; i < size; i++){ cout<< arr[i] << " "; } return 0; }输出结果

如果我们运行上面的代码,它将生成以下输出

排列前的数组: 0 3 2 1 5 4 Rearrangement of an array so that arr[i] becomes arr[arr[i]] with O(1) extra space is: 0 1 2 3 4 5