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

xiaoxiao2021-02-28  118

题目链接: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; }
转载请注明原文地址: https://www.6miu.com/read-74618.html

最新回复(0)