将 { 0, 1, 2, ..., N-1 } 的任意一个排列进行排序并不困难,这里加一点难度,要求你只能通过一系列的 Swap(0, *) —— 即将一个数字与 0 交换 —— 的操作,将初始序列增序排列。例如对于初始序列 { 4, 0, 2, 1, 3 },我们可以通过下列操作完成排序:
- Swap(0, 1) ⟹ { 4, 1, 2, 0, 3 }
- Swap(0, 3) ⟹ { 4, 1, 2, 3, 0 }
- Swap(0, 4) ⟹ { 0, 1, 2, 3, 4 }
本题要求你找出将前 N 个非负整数的给定排列进行增序排序所需要的最少的与 0 交换的次数。
输入格式:
输入在第一行给出正整数 N (≤105);随后一行给出{ 0, 1, 2, ..., N-1 } 的一个排列。数字间以空格分隔。
输出格式:
在一行中输出将给定序列进行增序排序所需要的最少的与 0 交换的次数。
输入样例:
10
3 5 7 2 6 4 9 0 8 1
输出样例:
9
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 2e5 + 10;
ll n;
ll ans, idx, flag, step;
ll a[N]; //记录每个点的数字
ll vis[N]; //记录这个点是否出现在其他环过
int main() {
cin >> n;
for (ll i = 0; i < n; i++)cin >> a[i];
for (ll i = 0; i < n; i++) {
if (!vis[i] && a[i] != i) {//如果这个点没有在其他环过,并且该位置的数字没有对应
flag = 0;//判断这个环里是否有0
idx = i;//环的起点
step = 0;//记录步数
while (!vis[idx]) {
if (!a[idx])flag = 1;//环里有0,则 flag 标注为 1
vis[idx] = 1;//标记这个点跑过
idx = a[idx];//更新idx为该环的下一个值
step++;
}
if (flag)ans += step - 1;//如果这个环有0,则不需要从别处调用0这步
else ans += step + 1;//反之需要调用0进入环
} else vis[i] = 1;//将值已经在原位的标记
}
cout << ans << endl;
}