CodeForces 814B An express train to reveries

题目链接：http://codeforces.com/contest/814/problem/B
题意：给你两个长度为n的序列a,b，让你求一个序列p，他满足，和a有一个元素不同，和b有一个元素不同，且是n的全排列，让你输入任意一种满足题意的序列p
解析：首先有两种情况，a和b有一个不同，那么那一个不同的应该就是n个元素里除去n-1个相同的剩下的，第二种情况就是a和b有两个不同，那么这两个位置就有两种情况，那么你把这两种情况都试一下就可以了（代码有点丑

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1005;
int a[maxn];
int b[maxn];
int c[maxn];
int vis[maxn];
int main(void)
{
    int n;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    for(int i=0;i<n;i++)
        scanf("%d",&b[i]);
    memset(vis,0,sizeof(vis));
    vector<int>tmp;
    int cnt = 0;
    for(int i=0;i<n;i++)
    {
        if(a[i]==b[i])
        {
            c[i] = a[i];
            vis[a[i]] = 1;
        }
        else
        {
            cnt++;
            tmp.push_back(i);
        }
    }
    if(cnt==1)
    {
        for(int i=0;i<n;i++)
        {
            if(c[i]==0)
            {
                for(int j=1;j<=n;j++)
                {
                    if(vis[j]==0)
                    {
                        c[i] = j;
                        break;
                    }
                }
                break;
            }
        }
    }
    else
    {
        int cnt1 = 0,cnt2 = 0;
        for(int i=1;i<=n;i++)
        {
            if(!vis[i])
                tmp.push_back(i);
        }
        c[tmp[0]] = tmp[2],c[tmp[1]] = tmp[3];
        for(int i=0;i<n;i++)
        {
            if(a[i]!=c[i])
                cnt1++;
            if(b[i]!=c[i])
                cnt2++;
        }
        if(cnt1==cnt2 && cnt1==1);
        else
            c[tmp[0]] = tmp[3],c[tmp[1]] = tmp[2];
    }
    for(int i=0;i<n;i++)
        printf("%d ",c[i]);
    puts("");
    return 0;
}