結果

問題 No.1051 PQ Permutation
コンテスト
ユーザー tarattata1
提出日時 2020-05-08 23:22:47
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,147 bytes
コンパイル時間 1,393 ms
コンパイル使用メモリ 90,032 KB
実行使用メモリ 14,592 KB
最終ジャッジ日時 2024-07-05 20:04:00
合計ジャッジ時間 5,894 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 TLE * 1 -- * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <string>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <iterator>
#include <cassert>
#include <numeric>
#include <functional>
#include <cassert>
//#include <numeric>
#pragma warning(disable:4996) 
 
typedef long long ll;
typedef unsigned long long ull;
#define MIN(a, b) ((a)>(b)? (b): (a))
#define MAX(a, b) ((a)<(b)? (b): (a))
#define LINF  9223300000000000000
#define LINF2 1223300000000000000
#define INF 2140000000
const long long MOD = 1000000007;
//const long long MOD = 998244353;

using namespace std;

bool solve0(int n, int p, int q, vector<int>& a)
{
    next_permutation(a.begin(), a.end());

    int flag=0, idp = -1, idq = -1;
    int i;
    for (i = 0; i < n; i++) {
        if (a[i] != i) flag = 1;
        if (a[i] == p) idp = i;
        if (a[i] == q) idq = i;
    }
    if (flag == 0) {
        return false;
    }
    if (idp < idq) {
        return true;
    }

    set<int> z;
    for (i = idq+1; i < n; i++) {
        z.insert(a[i]);
    }
    int curr = idq;
    while (1) {
        if (curr < 0) {
            return false;
        }
        auto it = z.rbegin();
        if (*it > idq) {
            break;
        }
        z.insert(a[curr]);
        curr--;
    }

    for (i = curr + 1; i < n; i++) {
        auto it=z.rbegin();
        a[i] = *it;
        z.erase(*it);
    }
    bool ok = solve0(n, p, q, a);
    return ok;
}

void solve()
{
    int n, p, q;
    scanf("%d%d%d", &n, &p, &q);  p--; q--;
    vector<int> a(n);
    int i;
    for (i = 0; i < n; i++) {
        scanf("%d", &a[i]); a[i]--;
    }
    bool ok = solve0(n, p, q, a);
    if (!ok) {
        printf("-1\n"); return;
    }
    for (i = 0; i < n; i++) {
        printf("%d", a[i]+1);
        if (i < n - 1) printf(" ");
        else printf("\n");
    }

    return;
}

int main(int argc, char* argv[])
{
#if 1
    solve();
#else
    int T;
    scanf("%d", &T);
    int t;
    for(t=0; t<T; t++) {
        //printf("Case #%d: ", t+1);
        solve();
    }
#endif
    return 0;
}
0