Codeforces Round #418 (Div. 2) B. An express train to reveries

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

题意：给出a、b两个数字序列和长度n，求一个序列，要求1到n每个数字都出现一次且与a、b序列分别有且仅有一处数字不同，若有多解，输出任意一个答案。

思路：读题读的我好辛苦Orz。。。
可以看出，a、b两序列相同的部分就是答案序列，不同的部分可能有两种情况：
（1）只有一个位置不同
统计相同的那部分1到n有哪个数字没出现过，直接填上即可。
（2）有两个位置不同
那同时也就有两个数字没用过，反正一共两种情况，放进去看看满不满足条件判断下就好。

代码：c++

#include <cstdio>#include <algorithm>#include <iostream>#include <vector>#include <set>using namespace std;const int maxn = 2000;int a[maxn], b[maxn];int ans[maxn];vector<int> pos;int main(){    int n;    cin >> n;    for(int i=0; i<n; i++)    {        scanf("%d", &a[i]);    }    for(int i=0; i<n; i++)    {        scanf("%d", &b[i]);    }    set<int> all;    for(int i=1; i<=n; i++)    {        all.insert(i);    }    for(int i=0; i<n; i++)    {        if(a[i]==b[i])        {            ans[i] = a[i];            all.erase(a[i]);        }        else        {            pos.push_back(i);        }    }    set<int>::iterator iter = all.begin();    if(pos.size()==1)    {        ans[pos[0]] = *iter;    }    else    {        int t1 = *iter;        iter++;        int t2 = *iter;        int p1 = pos[0];        int p2 = pos[1];        int differa = 0, differb = 0;        if(t1!=a[p1])        {            differa++;        }        if(t2!=a[p2])        {            differa++;        }        if(t1!=b[p1])        {            differb++;        }        if(t2!=b[p2])        {            differb++;        }        if(differa==1&&differb==1)        {            ans[p1] = t1;            ans[p2] = t2;        }        else        {            ans[p1] = t2;            ans[p2] = t1;        }    }    for(int i=0; i<n; i++)    {        if(i!=0)        {            putchar(' ');        }        printf("%d", ans[i]);    }    return 0;}