PAT A1067交换数字

减小复杂度的trick:和正常思路A[i]=num反过来，用数组的下标存储值，用数组的值表示下标,写成A[num]=i,方便对值为0的元素的位置下标进行查询，降低复杂度，从O（N^2）到O（N），避免超时。

#include<cstdio>
#include<algorithm>
#include <iostream>
using namespace std;
const int maxn=100100;
int pos[maxn];
int main(){
    int n,ans=0;
    scanf("%d",&n);
    int left=n-1;
    int num;
    for(int i=0;i<n;i++){
        scanf("%d",&num);
        pos[num]=i;
        if(num==i&&i!=0){
            left--;
        }
    }
    int k=1;
    while(left>0){
        if(pos[0]==0){
            while(k<n){
                if(pos[k]!=k){
                    swap(pos[0],pos[k]);
                    ans++;
                    break;
                }
                k++;
            }        
        }
        while(pos[0]!=0){           
            swap(pos[pos[0]],pos[0]);
            //数组的下标存储值，值存储下标，翻过来，这样可以减少这一步找到 值为0的元素 的复杂度，从O（N^2）到O（N）
            ans++;
            left--;
                   }
          
    }
    printf("%d\n",ans);
    return 0;
}

pos[0]表示0在pos[0]号位，因此要将值为pos[0]的元素，即pos[pos[0]]与pos[0]交换