自然数数组的排序

自然数数组的排序

题目描述

给定一个长度为N的整形数组arr，其中有N个互不相等的自然数1-N。请实现arr的排序，但是不要把下标$0\sim N-1$位置上的数通过直接赋值的方式替换成$1\sim N$

[要求]

时间复杂度为$O(n)$，空间复杂度为$O(1)$

输入描述：

第一行有一个整数N。表示数组长度
接下来一行有N个互不相等的自然数1-N。

输出描述：

输出N个整数表示排序后的结果

示例1

输入

5
2 1 4 5 3

输出

1 2 3 4 5 

备注：

$1⩽N⩽10^5$

---

题解：

此题最终的状态是：a[i] = i + 1（下标从0开始）。我们可以从左往右遍历，对每个元素：

• 如果 a[i] == i + 1，则跳过；
• 如果 a[i] != i + 1，说明它不再自己的位置上，应在 a[i] - 1 位置上，将其与 a[ a[i] - 1 ]位置元素进行替换，由于替换后位置 i 的元素可能不是 i + 1，所以应继续替换，直到 a[i] == i + 1停止；

这样遍历完整个数组，最多交换 n-1 次，即可把所有元素交换到其相应的位置，时间复杂度$O(n)$，额外的空间复杂度$O(1)$。

代码：

#include <cstdio>
#include <algorithm>

using namespace std;

const int N = 100010;

int n;
int a[N];

int main(void) {
    scanf("%d", &n);
    for ( int i = 0; i < n; ++i ) scanf("%d", a + i);
    for ( int i = 0; i < n; ++i ) {
        while ( a[i] != i + 1 ) swap( a[i], a[a[i] - 1]);
    }
    for ( int i = 0; i < n; ++i) printf("%d ", a[i]);
    return 0;
}