結果

問題 No.1311 Reverse Permutation Index
ユーザー Nzt3Nzt3
提出日時 2020-12-09 00:42:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 1,500 ms
コード長 776 bytes
コンパイル時間 1,426 ms
コンパイル使用メモリ 170,076 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-18 22:25:50
合計ジャッジ時間 1,873 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

const int mod=998244353;

int main(){
  int64_t n,s;
  cin>>n>>s;
  vector<int64_t>f(s);
  f[0]=1;
  for(int i=1;i<s;i++){
    f[i]=f[i-1]*i;
  }
  vector<int>a(s),p(s);
  iota(p.begin(),p.end(),1);
  for(int i=0;i<s-1;i++){
    int t=n/f[s-1-i],ind=0;
    n%=f[s-1-i];
    for(int j=0;j<s;j++){
      if(ind==t&&p[j]){ind=j;break;}
      else if(p[j])ind++;
    }
    a[i]=p[ind],p[ind]=0;
  }
  for(int i=0;i<s;i++)if(p[i])a[s-1]=p[i];
  vector<int>b(s);
  for(int i=0;i<s;i++){
    b[a[i]-1]=i+1;
  }
  int64_t ans=0;
  iota(p.begin(),p.end(),1);
  for(int i=0;i<s;i++){
    int ind=0;
    for(int j=0;j<s;j++){
      if(p[j]==b[i]){p[j]=0;break;}
      else if(p[j])ind++;
    }
    ans+=f[s-1-i]*ind;
  }
  cout<<ans<<'\n';
}
0