自然数数组的排序
题目描述
给定一个长度为N的整形数组arr,其中有N个互不相等的自然数1-N。请实现arr的排序,但是不要把下标 0 ∼ N − 1 0 \sim N-1 0∼N−1位置上的数通过直接赋值的方式替换成 1 ∼ N 1 \sim N 1∼N
[要求]
时间复杂度为 O ( n ) O(n) O(n),空间复杂度为 O ( 1 ) O(1) O(1)
输入描述:
第一行有一个整数N。表示数组长度
接下来一行有N个互不相等的自然数1-N。
输出描述:
输出N个整数表示排序后的结果
示例1
输入
5
2 1 4 5 3
输出
1 2 3 4 5
备注:
1 ⩽ N ⩽ 1 0 5 1 \leqslant N \leqslant 10^5 1⩽N⩽105
题解:
此题最终的状态是: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(n) O(n),额外的空间复杂度 O ( 1 ) O(1) 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;
}