結果

問題 No.2250 Split Permutation
ユーザー Nzt3Nzt3
提出日時 2023-03-31 17:08:32
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 66 ms / 3,000 ms
コード長 1,028 bytes
コンパイル時間 2,130 ms
コンパイル使用メモリ 197,404 KB
最終ジャッジ日時 2025-02-11 19:36:42
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
const int mod=998244353;

long long modpow(int a,int n){
  long long ret=1,t=a;
  for(int i=0;i<30;i++){
    if(n>>i&1)ret=ret*t%mod;
    t=t*t%mod;
  }
  return ret;
}
void add(int p,int v,vector<long long>& BIT){
  for(;p<BIT.size();p+=p&-p){
    BIT[p]=(BIT[p]+v)%mod;
  }
}
long long getsum(int l,int r,vector<long long>& BIT){
  long long ret=0;
  for(--l;l>0;l-=l&-l)ret=(ret-BIT[l])%mod;
  for(;r>0;r-=r&-r)ret=(ret+BIT[r])%mod;
  return ret;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin>>N;
  vector<int>P(N);
  for(int &i:P)cin>>i;
  long long ans=0;
  const long long inv2=modpow(2,mod-2);
  vector<long long>v2(N),rv2(N),cnt(N+1),mi(N+1);
  v2[0]=rv2[0]=1;
  for(int i=1;i<N;i++){
    v2[i]=v2[i-1]*2%mod;
    rv2[i]=rv2[i-1]*inv2%mod;
  }
  for(int i=0;i<N;i++){
    ans=(ans+(getsum(P[i],N,cnt)*v2[i]%mod-getsum(P[i],N,mi))%mod*v2[N-1-i])%mod;
    add(P[i],1,cnt);
    add(P[i],v2[i],mi);
  }
  if(ans<0)ans+=mod;
  cout<<ans<<'\n';
}
0