#946638 (C++17) No.2616 中央番目の中央値

• 問題: No.2616 中央番目の中央値
• ユーザー: inksamurai
• 提出日時: 2024-01-26 22:23:23
• 言語: C++17 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 189 ms / 2,000 ms
• コード長: 2,371 bytes
• コンパイル時間: 3,967 ms
• コンパイル使用メモリ: 254,112 KB
• 最終ジャッジ日時: 2025-02-18 23:27:05
• ジャッジサーバーID （参考情報）: judge3 / judge1

ファイルパターン	結果
other	AC * 37

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define rng(i,c,n) for(int i=c;i<n;i++)
#define fi first
#define se second
#define pb push_back
#define all(a) a.begin(), a.end()
#define sz(a) ((int) a.size())
#define vec(...) vector<__VA_ARGS__>
#define _3PawRSC ios::sync_with_stdio(0),cin.tie(0)
typedef long long ll;
using vi=vector<int>;
using pii=pair<int,int>;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}
// snuke's mod int
template <ll mod>
struct modint{
    ll x;
    modint(ll x=0):x((x%mod+mod)%mod){}
    modint operator-()const{return modint(-x);}
    modint& operator+=(const modint a){if((x+=a.x)>=mod) x-=mod; return *this;}
    modint& operator-=(const modint a){if((x+=mod-a.x)>=mod) x-=mod; return *this;}
    modint& operator*=(const modint a){(x*=a.x)%=mod; return *this;}
    modint operator+(const modint a)const{modint res(*this); return res+=a;}
    modint operator-(const modint a)const{modint res(*this); return res-=a;}
    modint operator*(const modint a)const{modint res(*this); return res*=a;}
    modint pow(ll n)const{
        modint res=1,x(*this);
        while(n){
            if(n&1)res*=x;
            x*=x;
            n>>=1;
        }
        return res;
    }
    modint inv()const{return pow(mod-2);}
};
using mint=modint<998244353>;
const int _n=3e5;
mint fact[_n],invfact[_n];
void initfact(){
    fact[0]=1;
    rng(i,1,_n){
        fact[i]=fact[i-1]*mint(i);
    }
    invfact[_n-1]=fact[_n-1].inv();
    per(i,_n-1){
        invfact[i]=invfact[i+1]*mint(i+1);
    }
}
mint cnk(ll k,ll n){
    return k>n?0:fact[n]*invfact[k]*invfact[n-k];
}
int op(int l,int r){
    return l+r;
}
int e(){
    return 0;
}
void slv(){
    int n;
    cin>>n;
    vi a(n);
    rep(i,n){
        cin>>a[i];
    }
    rep(i,n){
        a[i]-=1;
    }
    vec(vi) cl(2,vi(n));
    {
        atcoder::segtree<int,op,e> seg(n);
        rep(i,n){
            int x=a[i];
            cl[0][i]=seg.prod(0,x);
            cl[1][i]=seg.prod(x,n);
            seg.set(x,1);
        }
    }
    vec(vi) cr(2,vi(n));
    {
        atcoder::segtree<int,op,e> seg(n);
        per(i,n){
            int x=a[i];
            cr[0][i]=seg.prod(0,x);
            cr[1][i]=seg.prod(x,n);
            seg.set(x,1);
        }   
    }
    mint ans=0;
    rep(i,n){
        int x=min(cl[1][i],cr[0][i]);
        mint wys=cnk(x,cl[1][i]+cr[0][i]);
        int y=min(cl[0][i],cr[1][i]);
        wys*=cnk(y,cl[0][i]+cr[1][i]);
        ans+=wys;
    }
    print(ans.x);
}
signed main(){
_3PawRSC;
    initfact();
    slv();
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX