Sort with Swap(0,*) (25)

题目连接

题目描述
Given any permutation of the numbers {0, 1, 2,…, N-1}, it is easy to sort them in increasing order. But what if Swap(0, *) is the ONLY operation that is allowed to use? For example, to sort {4, 0, 2, 1, 3} we may apply the swap operations in the following way:
Swap(0, 1) => {4, 1, 2, 0, 3}
Swap(0, 3) => {4, 1, 2, 3, 0}
Swap(0, 4) => {0, 1, 2, 3, 4}
Now you are asked to find the minimum number of swaps need to sort the given permutation of the first N nonnegative integers.

输入描述:
Each input file contains one test case, which gives a positive N (<=105) followed by a permutation sequence of {0, 1, …, N-1}. All the numbers in a line are separated by a space.

输出描述:
For each case, simply print in a line the minimum number of swaps need to sort the given permutation.

输入例子:
10 3 5 7 2 6 4 9 0 8 1

输出例子:
9

---

大意：
给你一个整数数组，问若每次操作只能交换0和其他一个数组的位置，则最少需要多少次操作使得将这个数组
变成升序的数组。

分析
这是一条挺有意思的题目。我的第一思路是想直接找规律，但是找了很久后发现这个规律没有想象那么明显。
在题目描叙你们的数据样例只要从0开始，不断将数字放回本应在的位置，当0放到0时刚好全部排序好。但是后
来发现原来不是全部数据都行得通。这时引入了环的概念，假如没有规定只能用0和其他数字交换，那么按照上面的
方法，一个循环下来n次交换能够使n个数字回到正确的位置。仔细观察，只能用0和不做规定的差别就是操作次数的
不同。那么可以按照这个规律，找出数组中按照这个排序规则形成的‘环’，即可计算得操作次数。

MyCode:

#include<iostream>
#include<stdio.h>
#include<string>
using namespace std;
const int maxn = 100009;
int index[maxn];	//值为maxn的节点所在的下标
int val[maxn];		//下标为maxn的节点当前存放的值
bool ok[maxn];		//用于标记节点是否位于正确位置

//从下标为i的节点开始不断往下挖，直至出现环，返回挖出的节点数量
int count(int i){
	bool zero = false;
	int num = 0;
	while (ok[i] == false){
		if (val[i] == 0) zero = true;
		ok[i] = true;
		num++;
		i = index[i];
	}
	//如果环中有0，那么交换次数是节点数-1，否则是节点数+1
	return zero ? num - 1 : num + 1;
}

int main(){
	//get input data
	int n,ti;
	cin >> n;
	for (int i = 0; i < n; i++){
		scanf("%d", &ti);
		index[ti] = i;
		val[i] = ti;
		ok[i] = (ti == i ? true : false);
	}
	//caculate the answer
	int ans = 0;
	for (int i = 0; i < n; i++){
		if (ok[i] == false)
			ans += count(i);
	}
	cout << ans << endl;
}