結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 213 ms
42,624 KB
testcase_01 AC 294 ms
42,584 KB
testcase_02 AC 292 ms
42,624 KB
testcase_03 AC 300 ms
42,588 KB
testcase_04 AC 293 ms
42,720 KB
testcase_05 AC 306 ms
42,496 KB
testcase_06 AC 294 ms
42,624 KB
testcase_07 AC 297 ms
42,624 KB
testcase_08 AC 299 ms
42,624 KB
testcase_09 AC 293 ms
42,712 KB
testcase_10 AC 298 ms
42,584 KB
testcase_11 AC 160 ms
35,840 KB
testcase_12 AC 166 ms
35,968 KB
testcase_13 AC 245 ms
40,448 KB
testcase_14 AC 230 ms
39,680 KB
testcase_15 AC 278 ms
41,600 KB
testcase_16 AC 121 ms
28,800 KB
testcase_17 AC 31 ms
27,008 KB
testcase_18 AC 44 ms
28,288 KB
testcase_19 AC 294 ms
42,368 KB
testcase_20 AC 223 ms
39,168 KB
testcase_21 AC 179 ms
36,864 KB
testcase_22 AC 46 ms
28,672 KB
testcase_23 AC 151 ms
35,200 KB
testcase_24 AC 130 ms
34,176 KB
testcase_25 AC 69 ms
30,336 KB
testcase_26 AC 35 ms
28,032 KB
testcase_27 AC 251 ms
40,192 KB
testcase_28 AC 171 ms
36,352 KB
testcase_29 AC 259 ms
40,576 KB
testcase_30 AC 91 ms
31,744 KB
testcase_31 AC 79 ms
30,848 KB
testcase_32 AC 192 ms
37,632 KB
testcase_33 AC 134 ms
34,432 KB
testcase_34 AC 112 ms
33,024 KB
testcase_35 AC 135 ms
34,688 KB
testcase_36 AC 108 ms
32,896 KB
testcase_37 AC 227 ms
39,808 KB
testcase_38 AC 26 ms
26,880 KB
testcase_39 AC 25 ms
27,008 KB
testcase_40 AC 27 ms
27,008 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