Codeforces - 814B - An express train to reveries - 构造

http://codeforces.com/problemset/problem/814/B

构造题烦死人，一开始我还记录一大堆信息来构造p数列，其实因为s数列只有两项相等，也正好缺了一项，那就把两种情况构造出来暴力验证对不对就行了。

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

int n;
int s[];
int us[];
int t[];
int ut[];
int a[];

int main(){
    scanf("%d",&n);
    for(int i=;i<n;i++){
        scanf("%d",&s[i]);
        us[s[i]]++;
    }
    for(int i=;i<n;i++){
        scanf("%d",&t[i]);
        ut[t[i]]++;
    }

    int t1=;
    for(int i=;i<=n;i++){
        if(us[i]==){
            t1=i;
        }
    }

    vector<int> dif;
    for(int i=;i<n;i++){
        a[i]=s[i];
        if(us[s[i]]==){
            dif.push_back(i);
            //printf("i=%d\n",i);
        }
    }

    int tmp;
    for(auto i:dif){
        tmp=a[i];
        a[i]=t1;
        int cnt=;
        for(int j=;j<n;j++){
            if(a[j]!=t[j])
                cnt++;
        }
        if(cnt==){
            for(int j=;j<n;j++){
                printf("%d%c",a[j],(" \n"[j==n-]));
            }
            return ;
        }
        a[i]=tmp;
    }

}