結果

問題 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
コンパイル時間 859 ms
コンパイル使用メモリ 83,808 KB
実行使用メモリ 41,284 KB
最終ジャッジ日時 2024-04-23 12:10:54
合計ジャッジ時間 11,120 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 304 ms
41,088 KB
testcase_02 AC 289 ms
41,028 KB
testcase_03 AC 300 ms
40,960 KB
testcase_04 AC 287 ms
40,960 KB
testcase_05 AC 286 ms
41,088 KB
testcase_06 AC 311 ms
41,156 KB
testcase_07 AC 285 ms
41,028 KB
testcase_08 AC 319 ms
41,088 KB
testcase_09 AC 302 ms
41,088 KB
testcase_10 AC 301 ms
40,960 KB
testcase_11 AC 161 ms
35,072 KB
testcase_12 AC 165 ms
35,072 KB
testcase_13 AC 254 ms
39,296 KB
testcase_14 AC 231 ms
38,272 KB
testcase_15 AC 272 ms
40,064 KB
testcase_16 AC 120 ms
27,904 KB
testcase_17 AC 26 ms
27,008 KB
testcase_18 AC 42 ms
28,160 KB
testcase_19 AC 288 ms
40,832 KB
testcase_20 AC 222 ms
38,144 KB
testcase_21 AC 174 ms
35,712 KB
testcase_22 AC 47 ms
28,544 KB
testcase_23 AC 153 ms
34,432 KB
testcase_24 AC 132 ms
33,280 KB
testcase_25 AC 69 ms
29,952 KB
testcase_26 AC 36 ms
27,904 KB
testcase_27 AC 249 ms
38,912 KB
testcase_28 AC 168 ms
35,328 KB
testcase_29 AC 251 ms
39,040 KB
testcase_30 AC 90 ms
31,104 KB
testcase_31 AC 77 ms
30,464 KB
testcase_32 AC 191 ms
36,736 KB
testcase_33 AC 135 ms
33,792 KB
testcase_34 AC 113 ms
32,640 KB
testcase_35 AC 138 ms
33,792 KB
testcase_36 AC 114 ms
32,256 KB
testcase_37 AC 235 ms
38,784 KB
testcase_38 AC 27 ms
27,008 KB
testcase_39 AC 27 ms
27,136 KB
testcase_40 AC 26 ms
27,008 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