結果

問題 No.1193 Penguin Sequence
ユーザー pockynypockyny
提出日時 2020-08-22 18:56:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 417 ms / 2,000 ms
コード長 1,806 bytes
コンパイル時間 819 ms
コンパイル使用メモリ 84,864 KB
実行使用メモリ 42,496 KB
最終ジャッジ日時 2024-10-15 12:10:07
合計ジャッジ時間 12,762 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 255 ms
42,496 KB
testcase_01 AC 381 ms
42,496 KB
testcase_02 AC 385 ms
42,496 KB
testcase_03 AC 380 ms
42,496 KB
testcase_04 AC 379 ms
42,368 KB
testcase_05 AC 382 ms
42,496 KB
testcase_06 AC 398 ms
42,496 KB
testcase_07 AC 394 ms
42,368 KB
testcase_08 AC 393 ms
42,496 KB
testcase_09 AC 382 ms
42,368 KB
testcase_10 AC 382 ms
42,496 KB
testcase_11 AC 204 ms
35,712 KB
testcase_12 AC 212 ms
35,712 KB
testcase_13 AC 336 ms
40,320 KB
testcase_14 AC 317 ms
39,424 KB
testcase_15 AC 378 ms
41,600 KB
testcase_16 AC 141 ms
28,544 KB
testcase_17 AC 43 ms
26,880 KB
testcase_18 AC 61 ms
28,032 KB
testcase_19 AC 417 ms
42,240 KB
testcase_20 AC 309 ms
39,040 KB
testcase_21 AC 236 ms
36,608 KB
testcase_22 AC 64 ms
28,544 KB
testcase_23 AC 197 ms
35,072 KB
testcase_24 AC 165 ms
34,176 KB
testcase_25 AC 90 ms
30,080 KB
testcase_26 AC 52 ms
27,776 KB
testcase_27 AC 332 ms
40,064 KB
testcase_28 AC 217 ms
35,968 KB
testcase_29 AC 327 ms
40,320 KB
testcase_30 AC 116 ms
31,616 KB
testcase_31 AC 101 ms
30,720 KB
testcase_32 AC 262 ms
37,504 KB
testcase_33 AC 180 ms
34,304 KB
testcase_34 AC 145 ms
32,896 KB
testcase_35 AC 177 ms
34,304 KB
testcase_36 AC 143 ms
32,768 KB
testcase_37 AC 311 ms
39,808 KB
testcase_38 AC 42 ms
26,880 KB
testcase_39 AC 43 ms
26,880 KB
testcase_40 AC 43 ms
26,752 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
using namespace std;
typedef long long ll;
ll n,seg[400010];
void update(int p,ll val){
    for(seg[p += n] += val;p>1;p>>=1){
        seg[p>>1] = seg[p] + seg[p^1];
    }
}

ll query(int l,int r){
    ll res = 0;
    for(l += n,r += n; l<r;l>>=1,r>>=1){
        if(l&1) res += seg[l++];
        if(r&1) res += seg[--r];
    }
    return res;
}

ll a[200010],b[200010] = {};
map<int,int> mp;
constexpr ll mod = 998244353;
const int MX = 1000100;
ll f[MX],inv[MX],fi[MX];
void solve(){
    inv[1] = 1;
    for(int i=2;i<MX;i++){
        inv[i] = mod - (mod/i)*inv[mod%i]%mod;
    }
    f[0] = fi[0] = 1;
    for(int i=1;i<MX;i++){
        f[i] = f[i-1]*i%mod;
        fi[i] = fi[i-1]*inv[i]%mod;
    }
}

ll nck(ll n, ll k){
    if(n<0 || k<0 || n<k) return 0;
    return f[n]*fi[k]%mod*fi[n-k]%mod;
}

ll pw(ll a,ll x){
    a %= mod;
    ll ret = 1;
    while(x){
        if(x&1) (ret *= a) %= mod;
        x /= 2; (a *= a) %= mod;
    }
    return ret;
}

int main(){
    ll i; cin >> n;
    for(i=0;i<n;i++){
        cin >> a[i]; mp[a[i]] = 1;
    }
    int cnt = 0;
    for(auto x:mp){
        mp[x.first] = cnt; cnt++;
    }
    for(i=0;i<n;i++) a[i] = mp[a[i]];
    for(i=0;i<n;i++) b[a[i]]++;
    ll tento = 0,sum = 0,sum1 = 0;
    for(i=0;i<n;i++){
        update(a[i],1);
        tento += query(a[i] + 1,n);
    }
    for(i=n;i>=0;i--){
        (sum += b[i]*sum1%mod) %=mod; (sum1 += b[i]) %= mod;
    }
    solve();
    ll ans = 0;
    for(i=1;i<=n;i++) (ans += i*i%mod*(i - 1)%mod) %= mod;
    (ans *= sum*pw(2*n*n,mod - 2)%mod) %= mod;
    ll ans1 = 0;
    for(i=1;i<=n;i++) (ans1 += i*(i - 1)%mod) %= mod;
    (ans1 *= tento*pw(n*(n - 1),mod -2)%mod) %= mod;
    (ans += ans1) %= mod;
    for(i=1;i<=n;i++) (ans *= nck(n,i)) %= mod;
    cout << ans << endl;
}
0