結果

問題 No.1311 Reverse Permutation Index
ユーザー face4
提出日時 2020-12-08 09:04:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 1,189 bytes
コンパイル時間 557 ms
コンパイル使用メモリ 79,564 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-17 14:27:03
合計ジャッジ時間 1,058 ms
ジャッジサーバーID
(参考情報)
judge6 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<iostream>
#include<set>
#include<vector>
using namespace std;

typedef unsigned long long ull;

int main(){
    ull n;
    int s;
    cin >> n >> s;
    vector<ull> fact(21);
    fact[0] = 1;
    for(int i = 1; i <= 20; i++)    fact[i] = fact[i-1]*i;
    vector<int> origin(s);
    set<int> res;
    for(int i = 1; i <= s; i++) res.insert(i);
    ull acc = 0;
    for(int i = 0; i < s; i++){
        ull sum = 0, unit = fact[s-i-1];
        for(auto it = res.begin(); it != res.end(); it++){
            if(acc+sum+unit > n){
                origin[i] = *it;
                res.erase(it);
                acc += sum;
                break;
            }
            sum += unit;
        }
    }
    vector<int> rev(s);
    for(int i = 0; i < s; i++)  rev[origin[i]-1] = i+1;
    for(int i = 1; i <= s; i++) res.insert(i);
    ull ans = 0;
    for(int i = 0; i < s; i++){
        ull sum = 0, unit = fact[s-1-i];
        for(auto it = res.begin(); it != res.end(); it++){
            if(*it == rev[i]){
                ans += sum;
                res.erase(it);
                break;
            }
            sum += unit;
        }
    }
    cout << ans << endl;
    return 0;
}
0