結果

問題 No.1311 Reverse Permutation Index
ユーザー umezoumezo
提出日時 2020-12-08 00:54:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 1,500 ms
コード長 870 bytes
コンパイル時間 2,508 ms
コンパイル使用メモリ 195,296 KB
最終ジャッジ日時 2025-01-16 19:13:29
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

ll f(ll x){
  ll tmp=1;
  for(int i=2;i<=x;i++){
    tmp*=i;
  }
  return tmp;
}

int main(){
  ll n,s;
  cin>>n>>s;
  
  ll m=n;
  vector<ll> A(s+1);
  for(int i=s;i>0;i--){
    A[i]=m/f(i-1)+1;
    m-=m/f(i-1)*f(i-1);
  }
  
  vector<ll> B(s+1),C(s+1,1);
  for(int i=s;i>0;i--){
    int cnt=0;
    for(int j=1;j<=s;j++){
      if(C[j]==1) cnt++;
      if(A[i]==cnt){
        B[i]=j;
        C[j]=0;
        break;
      }
    }
  }
  
  vector<ll> D(s+1);
  for(int i=s;i>0;i--) D[B[i]]=s+1-i;
  
  ll ans=0;
  vector<ll> E(s+1,1);
  for(int i=1;i<=s;i++){
    ll cnt=0;
    for(int j=1;j<=D[i];j++){
      if(E[j]==1) cnt++;
    }
    ans+=f(s-i)*(cnt-1);
    E[D[i]]=0;
  }
  cout<<ans<<endl;
  
  return 0;
}
0