题目大意:
有一串数字an,我们对其进行长度为3的滑窗,滑窗每到一个地方就输出对应位置的3个数字。每次输出得到的3个数字为1组,我们叫它为一个滑窗。现在我们可以对滑窗内部的数字进行任意交换,另外我们还可以任意交换滑窗的位置。
现在我们有一系列进行过上面操作的滑窗,问我们是否能够还原出原来的数字an。
解题思路:
通过观察,我们知道an头和尾在滑窗中只会出现一次,我们把这个作为起点(我们可以知道这个新构成的数列可以和原来的首尾进行互调)。而头节点的下个数字只会在所有滑窗中出现两次,而且这个数字必须和头节点同一个滑窗。通过这个条件我们又找到了下一个点。接下来我们可以用stl的map<pair<int,int>,vector<int>>的结构来做到O(n log n )的复杂度添加答案。其中字典的key是我们当前已知的两个数,第三个数是可能出现的数,我们可以再建一个数组done来去重,防止重复拿数字。具体可以看代码。
废话:哎,做的时候没考虑清楚map的key值有多少个,以为有n^2个所以放弃了这种做法,其实key值只有2n个。。。
另外这题也复习了一下stl吧
#include <bits/stdc++.h>
using namespace std;
const int MAXN=1e5+10;
int cnt[MAXN];
vector<vector<int>> group(MAXN);
map<pair<int,int>,vector<int>> mm;
vector<int> ans;
vector<int> done(MAXN);
int main(){
int n;cin>>n;
for(int ii=0;ii<n-2;ii++){
int a[3];
for(int i=0;i<3;i++){cin>>a[i],cnt[a[i]]++;group[ii].emplace_back(a[i]);;}
sort(a,a+3);
mm[make_pair(a[0],a[1])].push_back(a[2]);
mm[make_pair(a[0],a[2])].push_back(a[1]);
mm[make_pair(a[1],a[2])].push_back(a[0]);
}
int head;
for(int i=0;i<MAXN;i++){
if(cnt[i]==1){
head=i;
done[head]=1;
ans.push_back(i);
break;
}
}
int sec;
for(int i=0;i<MAXN;i++){
for(int j=0;j<(int)group[i].size();j++){
if(group[i][j]==head){
for(int k=0;k<3;k++){
if(cnt[group[i][k]]==2){
sec=group[i][k];
ans.push_back(sec);
done[sec]=1;
break;
}
}
}
}
}
for(int i=0;i<n-2;i++){
int ls=ans.size();
ls-=1;
pair<int,int> key=make_pair(min(ans.back(),ans[ls-1]),max(ans.back(),ans[ls-1]));
for(auto it:mm[key]){
if(!done[it]){
done[it]=1;
ans.push_back(it);
}
}
}
for(auto it:ans){
cout<<it<<" ";
}
cout<<endl;
return 0;
}