10-排序6 Sort with Swap(0, i) (25分)

//表排序
/*
题意：给定N个数字的排列，利用与0的交换来排序，引理：
N个数字的排列有若干个独立的环组成，环分为三种：
1.只含一个元素：无需交换
2.环中有0：只需n-1次交换
3.环中无0：需要先把0换到环中，在进行n+1次交换
*/
#include <stdio.h>
#define maxn 100005

int a[maxn], table[maxn];

int main() {
    int N, temp = 0, x = 0, ans = 0;
    int i, j, cnt = 0, flag = 0;
    scanf("%d", &N);
    for(i = 0; i < N; i++) {
        scanf("%d", &a[i]);
        table[a[i]] = i;
    }
    for(i = 0; i < N; i++, cnt = 0) {
        if(table[i] != i) {
            j = i;
            temp = a[i];
        }
        while(table[j] != j) {
            if(a[j] == 0) flag = 1;
            x = j;
            j = table[j];
            a[x] = a[j];
            table[x] = x;
            cnt++;
        }
        a[x] = temp;
        if(cnt) {
            if(flag) { ans += cnt-1; flag = 0; }
            else ans += cnt+1;
        }
    }
    printf("%d\n", ans);
}