2928 你缺什么

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define N 10009
int a[N],b[N],ans[N],vis[N],n;
void out(){
    for(int i=1;i<=n;i++)
        printf("%d ",b[i]);
    puts("");
}
void dfs(int i){
    if(i<=n)
        for(int j=1;j<=n;j++)
            if(!vis[j]&&a[j]!=ans[i-1]){
                b[i]=j;//记录答案 
                ans[i]=a[j];//记录a之前的一个人的答案
                vis[j]=1;
                if(i==n){
                    out();exit(0);//只输出一次，不能是return 
                }
                dfs(i+1);
                vis[j]=0;
            }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",a+i);
    dfs(1);
    return 0;
}