CF_An express train to reveries 【模拟】

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

题目传送门

题目描述：

  给定一个数n，给两个序列a和b，a和b中包含n个数。要求你找出一个序列，分别和a，b两个序列都有且仅有一个不一样的元素。并且序列中的元素是从1~n所有的数，并且这样的数一定存在。

解题思路：

  因为有且仅有一个不一样的元素，所以a与b序列中最多会有两个元素不一样。因此这道题可分为两种情况。第一种，有一个不一样，找出那个不一样的元素的下标，将那个数替换成1~n没有出现的数即可；第二种，有两个不一样，他只可能有两种不一样的情况就是保留对角线的数，然后判断序列是不是1~n不重复的即可。

代码部分：

#include <iostream>#include <stdio.h>#include <string.h>#include <set>#include <map>#include <algorithm>#include <string>using namespace std;int main(){    int n;    int a[2005],b[2005];    while(~scanf("%d",&n))    {        int c[1005],d[1005];        map<int,int>M;        for(int i=1; i<=n; i++)            scanf("%d",&a[i]);        for(int i=1; i<=n; i++)            scanf("%d",&b[i]);        int k=0;        for(int i=1; i<=n; i++)        {            if(a[i]==b[i])            {                c[i]=a[i];                M[a[i]]++;            }            else            {                d[k++]=i;            }        }        if(k==1)        {            int s;            for(int i=1;i<=n;i++)            {                if(M[i]==0)                    s=i;            }            c[d[0]]=s;        }        else        {            int t1=d[0];            int t2=d[1];            if(M[a[t1]]==0&&M[b[t2]]==0)            {                c[t1]=a[t1];                c[t2]=b[t2];            }            else            {                c[t1]=b[t1];                c[t2]=a[t2];            }        }        for(int i=1; i<=n; i++)        {            if(i==1)                printf("%d",c[i]);            else                printf(" %d",c[i]);        }        printf("\n");    }    return 0;}