結果

問題 No.1193 Penguin Sequence
ユーザー chineristACchineristAC
提出日時 2020-08-22 14:54:34
言語 PyPy3
(7.3.13)
結果
AC  
実行時間 516 ms / 2,000 ms
コード長 1,612 bytes
コンパイル時間 1,280 ms
コンパイル使用メモリ 87,184 KB
実行使用メモリ 182,044 KB
最終ジャッジ日時 2023-08-05 12:09:41
合計ジャッジ時間 15,147 ms
ジャッジサーバーID
(参考情報)
judge12 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 283 ms
181,244 KB
testcase_01 AC 462 ms
182,044 KB
testcase_02 AC 452 ms
181,328 KB
testcase_03 AC 459 ms
181,288 KB
testcase_04 AC 445 ms
181,608 KB
testcase_05 AC 453 ms
181,216 KB
testcase_06 AC 452 ms
181,240 KB
testcase_07 AC 449 ms
181,604 KB
testcase_08 AC 448 ms
181,948 KB
testcase_09 AC 452 ms
181,372 KB
testcase_10 AC 441 ms
181,748 KB
testcase_11 AC 274 ms
117,168 KB
testcase_12 AC 270 ms
117,848 KB
testcase_13 AC 380 ms
163,724 KB
testcase_14 AC 359 ms
153,744 KB
testcase_15 AC 422 ms
180,112 KB
testcase_16 AC 199 ms
112,272 KB
testcase_17 AC 83 ms
80,832 KB
testcase_18 AC 125 ms
86,640 KB
testcase_19 AC 516 ms
181,320 KB
testcase_20 AC 364 ms
153,104 KB
testcase_21 AC 295 ms
120,952 KB
testcase_22 AC 131 ms
88,960 KB
testcase_23 AC 258 ms
116,592 KB
testcase_24 AC 231 ms
114,860 KB
testcase_25 AC 164 ms
99,100 KB
testcase_26 AC 114 ms
83,432 KB
testcase_27 AC 376 ms
161,800 KB
testcase_28 AC 289 ms
121,440 KB
testcase_29 AC 372 ms
163,476 KB
testcase_30 AC 193 ms
110,520 KB
testcase_31 AC 175 ms
104,156 KB
testcase_32 AC 313 ms
130,092 KB
testcase_33 AC 242 ms
115,044 KB
testcase_34 AC 221 ms
118,748 KB
testcase_35 AC 245 ms
114,516 KB
testcase_36 AC 215 ms
118,408 KB
testcase_37 AC 362 ms
154,212 KB
testcase_38 AC 85 ms
80,780 KB
testcase_39 AC 85 ms
80,668 KB
testcase_40 AC 83 ms
80,604 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class BIT():
    def __init__(self,n):
        self.BIT=[0]*(n+1)
        self.num=n

    def query(self,idx):
        res_sum = 0
        while idx > 0:
            res_sum += self.BIT[idx]
            idx -= idx&(-idx)
        return res_sum

    #Ai += x O(logN)
    def update(self,idx,x):
        while idx <= self.num:
            self.BIT[idx] += x
            idx += idx&(-idx)
        return

def cmb(n, r, mod):#コンビネーションの高速計算 
    if ( r<0 or r>n ):
        return 0
    r = min(r, n-r)
    return g1[n] * g2[r] * g2[n-r] % mod

mod = 998244353 #出力の制限
N = 2*10**5
g1 = [1]*(N+1) # 元テーブル
g2 = [1]*(N+1) #逆元テーブル
inverse = [1]*(N+1) #逆元テーブル計算用テーブル

for i in range( 2, N + 1 ):
    g1[i]=( ( g1[i-1] * i ) % mod )
    inverse[i]=( ( -inverse[mod % i] * (mod//i) ) % mod )
    g2[i]=( (g2[i-1] * inverse[i]) % mod )
inverse[0]=0

N=int(input())
A=list(map(int,input().split()))
S=list(set(A))
S.sort()
comp={i:e+1 for e,i in enumerate(S)}
n=max(comp[i] for i in comp)

rev=0
bit=BIT(n)
for i in range(N):
    c=comp[A[i]]
    rev+=i-bit.query(c)
    bit.update(c,1)

A.sort()
Small=0
tmp=0
for i in range(1,N):
    if A[i]==A[i-1]:
        Small+=tmp
    else:
        tmp=i
        Small+=tmp


ans=0

S=1
for i in range(1,N+1):
    S=S*cmb(N,i,mod)
    S%=mod

const=(N*(N+1)//2)**2
const%=mod
for i in range(1,N+1):
    const-=i**2
    const%=mod

const=(const*pow(2,mod-2,mod))%mod
const=(const*inverse[N])%mod
const=(const*inverse[N])%mod

ans+=S*const*Small
ans%=mod
ans+=rev*S*(N+1)*inverse[3]
ans%=mod
print(ans)
0