結果

問題 No.2616 中央番目の中央値
ユーザー gucci0512gucci0512
提出日時 2024-01-26 22:40:31
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 4,617 bytes
コンパイル時間 4,464 ms
コンパイル使用メモリ 238,964 KB
実行使用メモリ 20,292 KB
最終ジャッジ日時 2024-09-28 08:37:22
合計ジャッジ時間 10,400 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 23 ms
20,292 KB
testcase_01 AC 23 ms
13,312 KB
testcase_02 AC 22 ms
13,292 KB
testcase_03 AC 22 ms
13,256 KB
testcase_04 AC 22 ms
13,256 KB
testcase_05 AC 22 ms
13,324 KB
testcase_06 AC 23 ms
13,220 KB
testcase_07 AC 22 ms
13,232 KB
testcase_08 AC 23 ms
13,228 KB
testcase_09 AC 22 ms
13,224 KB
testcase_10 AC 23 ms
13,292 KB
testcase_11 AC 23 ms
13,192 KB
testcase_12 AC 24 ms
13,232 KB
testcase_13 AC 37 ms
13,256 KB
testcase_14 AC 59 ms
13,288 KB
testcase_15 AC 165 ms
13,136 KB
testcase_16 AC 586 ms
13,520 KB
testcase_17 AC 1,287 ms
13,372 KB
testcase_18 TLE -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
権限があれば一括ダウンロードができます

ソースコード

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;
    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]++;
        }
        mint ans=0;
        // rep(j,0,2)rep(k,0,2)cout << cnt[j][k] << " ";
        // cout << endl;
        int M1=min(cnt[1][1],cnt[0][0]);
        int M2=min(cnt[0][1],cnt[1][0]);
        mint tmp1=0;
        rep(j,0,M1+1){
            tmp1+=nck(cnt[1][1],j)*nck(cnt[0][0],j);
        }
        mint tmp2=0;
        rep(j,0,M2+1){
            tmp2+=nck(cnt[0][1],j)*nck(cnt[1][0],j);
        }
        ans=tmp1*tmp2;
        // cout << ans.val() << endl;
        sum+=ans;

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