codeforces 814B An express train to reveries

题目链接：http://codeforces.com/contest/814/problem/B

题意：给你两个长度为n的序列a和序列b，让你构造一个长度为n的序列p，1到n的每个数字都包含，使得p与序列a和序列b在对应位置都只有一个不同。（a，b不相同，p一定存在）。

分析：a和b的相同位置如果元素相同，那么构造的p对于位置也是这个元素，我们将vis[a[i]]=1。我们接下来找相同位置的元素不同的位置下标，然后看1到n这些数字有哪些元素vis[]=0,等于0代表这些元素还没有出现。如果ab只有一个位置元素不同，那么就将未出现的数字填入，输出即可。如果有两个位置不同，要想构造的p与a和b各只有一个位置不同，假设没出现的两个数字是num1和num2，那么当a[ans1]==num1&&b[ans2]==num2||a[ans2]==num2&&b[ans1]=num1时p[ans1]=num1,p[ans2]=num2.其他情况反正即可。

AC代码：

#include<bits/stdc++.h>
using namespace std;

int a[1005],b[1005],c[1005],d[1005];
int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
    }
    for(int i=1;i<=n;i++){
        cin>>b[i];
    }
    int ans1=0,ans2=0;
    memset(d,0,sizeof(d));
    for(int i=1;i<=n;i++){
        if(a[i]==b[i]){
            c[i]=a[i];
            d[a[i]]=1;
        }
        else {
            if(ans1==0) ans1=i;
            else ans2=i;
        }
    }
    int num1=0,num2=0;
    for(int i=1;i<=n;i++){
        if(d[i]==0){
            if(num1==0) num1=i;
            else num2=i;
        }
    }
    if(ans2==0){
        c[ans1]=num1;
    }
    else {
        if(a[ans1]==num1&&b[ans2]==num2||a[ans2]==num2&&b[ans1]==num1){
            c[ans1]=num1;
            c[ans2]=num2;
        }
        else {
            c[ans1]=num2;
            c[ans2]=num1;
        }
    }
    for(int i=1;i<=n;i++){
        cout<<c[i]<<" ";
    }
    cout<<endl;
    return 0;
}