寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #ITP2_5_C : Permutation

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Permutation"
https://onlinejudge.u-aizu.ac.jp/problems/ITP2_5_C
与えらえた数列 A={a0,a1,...,an−1}について、辞書式順で前の順列と次の順列を求めてください。

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #ITP2_5_C : Permutation
/*
Aizu Online Judge in C++ #ITP2_5_C : Permutation
https://onlinejudge.u-aizu.ac.jp/problems/ITP2_5_C
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    int n;
    cin>>n;
    vector<int> a(n),x(n),y(n);
    for(int i=0;i<n;i++){
        cin>>a[i];
        x[i]=a[i];
        y[i]=a[i];
    }
    int b=0;
    while(prev_permutation(x.begin(),x.end()) && b!=1){
        for(int i=0;i<n;i++){
            cout<<x[i]<<((i==n-1) ? "\n" : " ");
        }
        b++;
    }
    for(int i=0;i<n;i++){
        cout<<a[i]<<((i==n-1) ? "\n" : " ");
    }
    int c=0;
    while(next_permutation(y.begin(),y.end()) && c!=1){
        for(int i=0;i<n;i++){
            cout<<y[i]<<((i==n-1) ? "\n" : " ");
        }
        c++;
    }
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/