Restoring the Permutation | Codeforces

Restoring the Permutation

---

​​from Codeforces Round #710 (Div. 3)​​

Time limit：2s

Memory limit：256MB

---

---

这个题目，题目读懂了就很简单，stl应用，我学到了

对于每一个测试，相同的数，第一个原型输出，后面的在比它小的数里面依次找最小的，第二个输出依次找最大的。

ac代码：

#include<iostream>
#include<set>
using namespace std;
int t,n;
int x[200005];
void solve(){
    set<int> s;
    for(int i = 1;i <= n;++i)
        s.insert(i);
    for(int i = 1;i <= n;++i)
        if(x[i] != x[i - 1])
            cout<<x[i]<<" ",s.erase(x[i]);
        else
            cout<<*s.begin()<<" ",s.erase(s.begin());
    cout<<"\n";
    for(int i = 1;i <= n;++i)
        s.insert(i);
    for(int i = 1;i <= n;++i)
        if(x[i] != x[i - 1])
            cout<<x[i]<<" ",s.erase(x[i]);
        else
            cout<<*prev(s.lower_bound(x[i]))<<" ",s.erase(*prev(s.lower_bound(x[i])));
    cout<<"\n";
}
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        for(int i = 1;i <= n;++i)
            cin>>x[i];
        solve();
    }
    return 0;
}