PAT1 1067 Sort with Swap(0, i)

题目链接
我的github

题目大意

给一个长度为$N$的数字序列，且序列中的每个数范围都在$[0,N-1]$，要求每次只能用0和其他数字进行交换，求最少需要多少次可以将这个序列变成一个升序序列

输入

每组包含一个测试用例

• 第一行是一个正数$N\le 10^5$表示序列的长度
• 第二行就是$N$个数字表示这个序列

输出

对每个测试用例输出交换的最小次数

样例输入

10
3 5 7 2 6 4 9 0 8 1

样例输出

9

解析

每次用0和最后应该在0在的位置上的数字进行交换，但是在交换的过程中有可能0跑到了位置0（0到了它最后应该在的位置）。
由于要求只能用0进行交换，所以就要继续用0将其他位上的数字交换到位置0，然后继续用最后应该在0在的位置上的数字进行交换

但是这题用python会出现返回非0的问题，不知道问题出哪了。。。
最后用C++AC的

python：

# -*- coding: utf-8 -*- 
# @Time : 2019/6/27 10:28 
# @Author : ValarMorghulis 
# @File : 1067.py
def solve():
    n = int(input())
    a = list(map(int, input().split()))
    t = [0 for i in range(n)]
    for i in range(n):
        t[a[i]] = i
    cnt = 0
    for i in range(n):
        while t[0] != 0:
            t[t[0]], t[0] = t[0], t[t[0]]
            cnt += 1
        if t[i] != i:
            t[i], t[0] = t[0], t[i]
            cnt += 1
    print(cnt)


if __name__ == "__main__":
    solve()

C++：

#include<stdio.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<stack>
#include<cstring>
#include<string>
#include<cmath>

#define inf 0xffffffff

using namespace std;

int main()
{
    int n;
    scanf("%d", &n);
    int t[n];
    for(int i=0; i<n; i++)
    {
        int x;
        scanf("%d", &x);
        t[x]=i;
    }
    int cnt=0;
    for(int i=0; i<n; i++)
    {
        while(t[0]!=0)
        {
            swap(t[t[0]], t[0]);
            cnt++;
        }
        if(t[i]!=i)
        {
            swap(t[i], t[0]);
            cnt++;
        }
    }
    printf("%d\n", cnt);
    return 0;
}