結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 267 ms
176,008 KB
testcase_01 AC 448 ms
177,988 KB
testcase_02 AC 457 ms
177,300 KB
testcase_03 AC 463 ms
177,604 KB
testcase_04 AC 465 ms
177,288 KB
testcase_05 AC 476 ms
177,800 KB
testcase_06 AC 467 ms
177,572 KB
testcase_07 AC 466 ms
177,768 KB
testcase_08 AC 472 ms
178,172 KB
testcase_09 AC 473 ms
177,676 KB
testcase_10 AC 470 ms
177,780 KB
testcase_11 AC 266 ms
124,984 KB
testcase_12 AC 271 ms
124,184 KB
testcase_13 AC 398 ms
161,340 KB
testcase_14 AC 352 ms
136,336 KB
testcase_15 AC 430 ms
174,228 KB
testcase_16 AC 185 ms
107,856 KB
testcase_17 AC 57 ms
64,804 KB
testcase_18 AC 104 ms
84,748 KB
testcase_19 AC 463 ms
177,616 KB
testcase_20 AC 358 ms
134,520 KB
testcase_21 AC 301 ms
120,704 KB
testcase_22 AC 111 ms
87,660 KB
testcase_23 AC 252 ms
129,880 KB
testcase_24 AC 226 ms
122,576 KB
testcase_25 AC 145 ms
98,264 KB
testcase_26 AC 94 ms
82,096 KB
testcase_27 AC 376 ms
142,256 KB
testcase_28 AC 273 ms
124,996 KB
testcase_29 AC 374 ms
141,992 KB
testcase_30 AC 176 ms
109,292 KB
testcase_31 AC 159 ms
103,188 KB
testcase_32 AC 316 ms
126,032 KB
testcase_33 AC 238 ms
127,628 KB
testcase_34 AC 202 ms
118,240 KB
testcase_35 AC 243 ms
127,968 KB
testcase_36 AC 199 ms
117,672 KB
testcase_37 AC 366 ms
136,008 KB
testcase_38 AC 56 ms
63,744 KB
testcase_39 AC 55 ms
63,616 KB
testcase_40 AC 56 ms
63,616 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