結果

問題 No.1193 Penguin Sequence
ユーザー pockynypockyny
提出日時 2020-08-22 18:49:41
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,883 bytes
コンパイル時間 931 ms
コンパイル使用メモリ 82,944 KB
実行使用メモリ 40,960 KB
最終ジャッジ日時 2024-10-15 12:09:53
合計ジャッジ時間 13,513 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 428 ms
40,960 KB
testcase_02 AC 434 ms
40,960 KB
testcase_03 AC 427 ms
40,960 KB
testcase_04 AC 423 ms
40,960 KB
testcase_05 AC 413 ms
40,960 KB
testcase_06 AC 419 ms
40,960 KB
testcase_07 AC 413 ms
40,832 KB
testcase_08 AC 421 ms
40,832 KB
testcase_09 AC 416 ms
40,832 KB
testcase_10 AC 421 ms
40,960 KB
testcase_11 AC 221 ms
34,688 KB
testcase_12 AC 226 ms
34,944 KB
testcase_13 AC 355 ms
38,912 KB
testcase_14 AC 332 ms
38,144 KB
testcase_15 AC 392 ms
40,064 KB
testcase_16 AC 143 ms
27,776 KB
testcase_17 AC 43 ms
26,752 KB
testcase_18 AC 61 ms
28,032 KB
testcase_19 AC 416 ms
40,704 KB
testcase_20 AC 308 ms
37,888 KB
testcase_21 AC 251 ms
35,712 KB
testcase_22 AC 66 ms
28,416 KB
testcase_23 AC 206 ms
34,176 KB
testcase_24 AC 173 ms
33,280 KB
testcase_25 AC 93 ms
29,696 KB
testcase_26 AC 52 ms
27,648 KB
testcase_27 AC 342 ms
38,656 KB
testcase_28 AC 233 ms
35,072 KB
testcase_29 AC 352 ms
39,040 KB
testcase_30 AC 118 ms
31,104 KB
testcase_31 AC 102 ms
30,336 KB
testcase_32 AC 262 ms
36,352 KB
testcase_33 AC 181 ms
33,664 KB
testcase_34 AC 148 ms
32,256 KB
testcase_35 AC 188 ms
33,664 KB
testcase_36 AC 145 ms
32,000 KB
testcase_37 AC 341 ms
38,528 KB
testcase_38 AC 44 ms
26,752 KB
testcase_39 AC 45 ms
26,752 KB
testcase_40 AC 44 ms
26,752 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
using namespace std;
typedef long long ll;
ll n,seg[400010],INF = 1000000000000;
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;
}
int 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 - 1;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) %= mod;
    (ans *= pw(2*n*n,mod - 2)) %= mod;
    ll ans1 = 0;
    for(i=1;i<=n;i++) (ans1 += i*(i - 1)%mod) %= mod;
    (ans1 *= tento) %= mod;
    (ans1 *= pw(n*(n - 1),mod -2)) %= mod;
    (ans += ans1) %= mod;
    ll ans2 = ans;
    for(i=1;i<=n;i++) (ans *= nck(n,i)) %= mod;
    cout << ans << endl;
}
0