結果

問題 No.1193 Penguin Sequence
ユーザー chineristACchineristAC
提出日時 2020-08-22 14:54:34
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 496 ms / 2,000 ms
コード長 1,612 bytes
コンパイル時間 705 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 178,264 KB
最終ジャッジ日時 2024-04-23 09:19:00
合計ジャッジ時間 15,537 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 278 ms
176,136 KB
testcase_01 AC 496 ms
178,264 KB
testcase_02 AC 484 ms
177,596 KB
testcase_03 AC 481 ms
177,592 KB
testcase_04 AC 480 ms
177,592 KB
testcase_05 AC 488 ms
177,468 KB
testcase_06 AC 479 ms
177,600 KB
testcase_07 AC 476 ms
177,732 KB
testcase_08 AC 488 ms
177,980 KB
testcase_09 AC 490 ms
177,724 KB
testcase_10 AC 492 ms
177,812 KB
testcase_11 AC 282 ms
125,164 KB
testcase_12 AC 279 ms
124,288 KB
testcase_13 AC 416 ms
161,180 KB
testcase_14 AC 388 ms
136,148 KB
testcase_15 AC 468 ms
174,276 KB
testcase_16 AC 191 ms
107,404 KB
testcase_17 AC 56 ms
63,616 KB
testcase_18 AC 105 ms
85,120 KB
testcase_19 AC 489 ms
177,372 KB
testcase_20 AC 379 ms
134,696 KB
testcase_21 AC 325 ms
121,036 KB
testcase_22 AC 111 ms
87,448 KB
testcase_23 AC 280 ms
129,536 KB
testcase_24 AC 248 ms
122,752 KB
testcase_25 AC 151 ms
98,304 KB
testcase_26 AC 100 ms
82,560 KB
testcase_27 AC 409 ms
141,812 KB
testcase_28 AC 300 ms
125,184 KB
testcase_29 AC 405 ms
141,872 KB
testcase_30 AC 182 ms
109,440 KB
testcase_31 AC 168 ms
103,040 KB
testcase_32 AC 330 ms
126,080 KB
testcase_33 AC 243 ms
127,616 KB
testcase_34 AC 218 ms
118,188 KB
testcase_35 AC 257 ms
127,992 KB
testcase_36 AC 214 ms
117,632 KB
testcase_37 AC 385 ms
136,072 KB
testcase_38 AC 56 ms
63,488 KB
testcase_39 AC 54 ms
63,744 KB
testcase_40 AC 55 ms
63,440 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