Codeforces Round #656 (Div. 3) B. Restore the Permutation by Merger

B. Restore the Permutation by Merger

题目链接-B. Restore the Permutation by Merger
Codeforces Round #656 (Div. 3) B. Restore the Permutation by Merger_第1张图片
Codeforces Round #656 (Div. 3) B. Restore the Permutation by Merger_第2张图片
题目大意
有一种排列 p p p,将第二个p的元素插入到第一个保持相对顺序的元素中,结果是长度为2n的序列 a a a,现给你插入后的序列 a a a,请你求出原序列 p p p

解题思路

  • 因为是保持相对顺序插入的,所以我们直接遍历序列 a a a找排列即可
  • 因为排列中不能有重复的元素,为了方便我们可以将已经输出的元素存入set,用s.count(a[i])判断已经确定的 p p p序列中是否已经有 a [ i ] a[i] a[i]元素,s.size()==n时说明 p p p排列中所有元素都已经确定,退出循环即可
  • 具体操作见代码

附上代码

#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include
#define int long long
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=2e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
int a[55];
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);

	int t;
	cin>>t;
	while(t--){
		int n;
		cin>>n;
		set<int> s;
		for(int i=1;i<=2*n;i++)
			cin>>a[i];
		for(int i=1;i<=2*n;i++){
			if(!s.count(a[i])){
				cout<<a[i]<<" ";
				s.insert(a[i]);
			}
			if(s.size()==n) break;
		}
		cout<<endl;
	}
	return 0;
}

你可能感兴趣的:(codeforces)