【Codeforces 814 B. An express train to reveries】+ 构造

B. An express train to reveries
time limit per test1 second
memory limit per test256 megabytes
inputstandard input
outputstandard output
Sengoku still remembers the mysterious “colourful meteoroids” she discovered with Lala-chan when they were little. In particular, one of the nights impressed her deeply, giving her the illusion that all her fancies would be realized.
On that night, Sengoku constructed a permutation p1, p2, …, pn of integers from 1 to n inclusive, with each integer representing a colour, wishing for the colours to see in the coming meteor outburst. Two incredible outbursts then arrived, each with n meteorids, colours of which being integer sequences a1, a2, …, an and b1, b2, …, bn respectively. Meteoroids’ colours were also between 1 and n inclusive, and the two sequences were not identical, that is, at least one i (1 ≤ i ≤ n) exists, such that ai ≠ bi holds.
Well, she almost had it all — each of the sequences a and b matched exactly n - 1 elements in Sengoku’s permutation. In other words, there is exactly one i (1 ≤ i ≤ n) such that ai ≠ pi, and exactly one j (1 ≤ j ≤ n) such that bj ≠ pj.
For now, Sengoku is able to recover the actual colour sequences a and b through astronomical records, but her wishes have been long forgotten. You are to reconstruct any possible permutation Sengoku could have had on that night.
Input
The first line of input contains a positive integer n (2 ≤ n ≤ 1 000) — the length of Sengoku’s permutation, being the length of both meteor outbursts at the same time.
The second line contains n space-separated integers a1, a2, …, an (1 ≤ ai ≤ n) — the sequence of colours in the first meteor outburst.
The third line contains n space-separated integers b1, b2, …, bn (1 ≤ bi ≤ n) — the sequence of colours in the second meteor outburst. At least one i (1 ≤ i ≤ n) exists, such that ai ≠ bi holds.
Output
Output n space-separated integers p1, p2, …, pn, denoting a possible permutation Sengoku could have had. If there are more than one possible answer, output any one of them.
Input guarantees that such permutation exists.
Examples
Input
5
1 2 3 4 3
1 2 5 4 5
Output
1 2 5 4 3
Input
5
4 4 2 3 1
5 4 5 3 1
Output
5 4 2 3 1
Input
4
1 1 3 4
1 4 3 4
Output
1 2 3 4
Note
In the first sample, both 1, 2, 5, 4, 3 and 1, 2, 3, 4, 5 are acceptable outputs.
In the second sample, 5, 4, 2, 3, 1 is the only permutation to satisfy the constraints.

1) 一个位置不同,把不同的位置替换为 a 数组里没有的数即可

2)两个位置不同,把这两个位置里重复的数替换掉

AC代码:

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX = 1e3 + 10;
typedef long long LL;
int a[MAX],b[MAX],c[MAX],vis[MAX];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i = 1; i <= n; i++)
        scanf("%d",&a[i]);
    for(int i = 1; i <= n; i++)
        scanf("%d",&b[i]);
    int p = 0,q = 0;
   for(int i = 1; i<= n; i++)
        if(a[i] != b[i]){
            if(!p)
                p = i;
            else
                q = i;
        }
     for(int i = 1; i <= n; i++)
            c[i] = a[i];
   if(!q){
        for(int i = 1; i <= n; i++)
            vis[a[i]] = 1;
        for(int i = 1; i <= n; i++)
            if(!vis[i])
                c[p] = i;
   }
   else{
       for(int i = 1; i <= n; i++)
            if(i != p && i != q)
                vis[a[i]] = 1;
        for(int i = 1; i <= n; i++)
            if(i != p && i != q)
                vis[b[i]] = 1;
        if(!vis[a[p]] && vis[b[p]]) c[p] = a[p],vis[c[p]] = 1;
        if(!vis[a[q]] && vis[b[q]]) c[q] = a[q],vis[c[q]] = 1;
        if(vis[a[q]] && !vis[b[q]]) c[q] = b[q],vis[c[q]] = 1;
        if(vis[a[p]] && !vis[b[p]]) c[p] = b[p],vis[c[p]] = 1;
        if(a[p] == a[q] && b[p] == b[q])
            c[q] = b[q];
   }
    for(int i = 1; i < n; i++)
        printf("%d ",c[i]);
    printf("%d\n",c[n]);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值