結果

問題 No.2616 中央番目の中央値
ユーザー gucci0512gucci0512
提出日時 2024-01-26 23:01:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 117 ms / 2,000 ms
コード長 4,690 bytes
コンパイル時間 4,290 ms
コンパイル使用メモリ 235,440 KB
実行使用メモリ 16,700 KB
最終ジャッジ日時 2024-09-28 08:51:32
合計ジャッジ時間 7,980 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 37
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
#pragma GCC optimize("Ofast")
#pragma GCC optimize("unroll-loops")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,fma,abm,mmx,avx,avx2")
// url 
#define rep(i,a,b) for(int i=a;i<b;i++)
#define rrep(i,a,b) for(int i=b-1;i>=a;i--)
#define all(x) (x).begin(),(x).end()
#define pb(x) push_back(x);
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a = b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a = b; return 1; } return 0; }
typedef long long ll;
typedef long double lld;
using namespace std;
using namespace atcoder;
using mint = static_modint<998244353>;
const ll mod=998244353;
// using mint = static_modint<1000000007>;
//const ll mod=1e9+7;
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
const string zton="0123456789";
const string atoz="abcdefghijklmnopqrstuvwxyz";
const string ATOZ="ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const ll inf=(1ll<<60);
// const int inf=(1<<30);
lld dist(lld x1,lld x2,lld y1,lld y2){
    lld res=(x1-x2)*(x1-x2)+(y1-y2)*(y1-y2);
    res=sqrt(abs(res));
    return res;
}
lld arg(lld x,lld y){
    const lld eps=1e-8;
    lld res=0;
    if(abs(x)+abs(y)<=eps)return 0.0;
    else if(abs(x)<=eps){
        if(y>=0.0)return (M_PI/2);
        else return (M_PI/2+M_PI);
    }
    else if(abs(y)<=eps){
        if(x>=0.0)return 0.0;
        else return M_PI;
    }
    res=atan2(abs(y),abs(x));
    if(x<=0&&y>=0)res=(M_PI-res);
    else if(x<=0&&y<=0)res+=(M_PI);
    else if(x>=0&&y<=0)res=(M_PI*2-res);
    return res;
}
ll gcd(ll a,ll b){
    if(a==0||b==0)return a+b;
    ll r;
    r=a%b;
    if(r==0){
        return b;
    }
    else{
        return gcd(b,r);
    }
}
std::ostream &operator<<(std::ostream &dest, __int128_t value) {
    std::ostream::sentry s(dest);
    if (s) {
        __uint128_t tmp = value < 0 ? -value : value;
        char buffer[128];
        char *d = std::end(buffer);
        do {
            --d;
            *d = "0123456789"[tmp % 10];
            tmp /= 10;
        } while (tmp != 0);
        if (value < 0) {
            --d;
            *d = '-';
        }
        int len = std::end(buffer) - d;
        if (dest.rdbuf()->sputn(d, len) != len) {
            dest.setstate(std::ios_base::badbit);
        }
    }
    return dest;
}
__int128 parse128(string &s) {
    __int128 ret = 0;
    for (int i = 0; i < s.length(); i++){
        if ('0' <= s[i] && s[i] <= '9'){
            ret = 10 * ret + s[i] - '0';
        }
    }
    return ret;
}
using l3 = __int128;

bool comp_arg(pair<ll,ll> a,pair<ll,ll> b){
    if(a.second>=0&&b.second<0)return true;
    else if(a.second<0&&b.second>=0)return false;
    if(a.second>=0&&b.second>=0){
        if(a.first>=0&&b.first<0)return true;
    }
    else{
        if(a.first>=0&&b.first<0)return false;
    }
    ll det=a.first*b.second-a.second*b.first;
    if(det>0)return true;
    return false;
}

typedef pair<ll,int> P;
const int kaijouSize=501010;
ll inv_[kaijouSize];
mint inv[kaijouSize];
mint kaijou[kaijouSize];
mint kaijou_inv[kaijouSize];
void InitKaijou(){
    inv_[0]=1;
    inv_[1]=1;
    inv[0]=1;
    inv[1]=1;
    rep(i,2,kaijouSize) inv_[i]=inv_[mod%i]*(mod-mod/i)%mod;
    rep(i,2,kaijouSize) inv[i]=inv_[i];
    kaijou[0]=1;
    rep(i,1,kaijouSize){
        kaijou[i]=kaijou[i-1]*1ll*i;
    }
    kaijou_inv[0]=1ll;
    rep(i,1,kaijouSize){
        kaijou_inv[i]=kaijou_inv[i-1]*1ll*(inv[i]);
    }
}
mint nck(int n,int k){
    if(n<k)return 0;
    else return (((kaijou[n]*kaijou_inv[k]))*kaijou_inv[n-k]);
}
mint npk(int n,int k){
    if(n<k)return 0;
    return (kaijou[n]*kaijou_inv[n-k]);
}

int main(void){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    InitKaijou();
    int N;cin >> N;
    vector<int> p(N);
    rep(i,0,N){
        cin >> p[i];
        p[i]--;
    }
    mint sum=0;
    fenwick_tree<int> ftr(N),ftl(N);rep(i,0,N)ftr.add(i,1);
    rep(i,0,N){
        // int cnt[2][2];
        // rep(j,0,2)rep(k,0,2)cnt[j][k]=0;
        // rep(l,0,i){
        //     if(p[l]<p[i])cnt[1][1]++;
        //     else cnt[0][1]++;
        // }
        // rep(l,i+1,N){
        //     if(p[l]<p[i])cnt[1][0]++;
        //     else cnt[0][0]++;
        // }
        int a=ftr.sum(p[i]+1,N);
        int b=ftl.sum(p[i]+1,N);
        int c=ftr.sum(0,p[i]);
        int d=ftl.sum(0,p[i]);
        mint ans=0;
        // int a=cnt[0][0],b=cnt[0][1],c=cnt[1][0],d=cnt[1][1];
        mint tmp1=kaijou[a+d]*kaijou_inv[a]*kaijou_inv[d];
        mint tmp2=kaijou[b+c]*kaijou_inv[b]*kaijou_inv[c];
        ans=tmp1*tmp2;
        // cout << ans.val() << endl;
        sum+=ans;
        ftr.add(p[i],-1);
        ftl.add(p[i],1);

    }
    cout << sum.val() << endl;
}
0