結果
問題 | No.1897 Sum of 2nd Max |
ユーザー | persimmon-persimmon |
提出日時 | 2022-04-14 14:04:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 671 ms / 2,000 ms |
コード長 | 1,475 bytes |
コンパイル時間 | 1,138 ms |
コンパイル使用メモリ | 86,308 KB |
実行使用メモリ | 124,160 KB |
最終ジャッジ日時 | 2023-08-25 20:40:39 |
合計ジャッジ時間 | 12,570 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
70,996 KB |
testcase_01 | AC | 72 ms
71,060 KB |
testcase_02 | AC | 71 ms
70,856 KB |
testcase_03 | AC | 610 ms
123,920 KB |
testcase_04 | AC | 599 ms
107,196 KB |
testcase_05 | AC | 319 ms
102,092 KB |
testcase_06 | AC | 295 ms
78,664 KB |
testcase_07 | AC | 416 ms
112,868 KB |
testcase_08 | AC | 337 ms
99,640 KB |
testcase_09 | AC | 599 ms
123,880 KB |
testcase_10 | AC | 569 ms
123,708 KB |
testcase_11 | AC | 576 ms
123,900 KB |
testcase_12 | AC | 576 ms
124,100 KB |
testcase_13 | AC | 584 ms
124,160 KB |
testcase_14 | AC | 187 ms
79,220 KB |
testcase_15 | AC | 191 ms
79,180 KB |
testcase_16 | AC | 252 ms
78,884 KB |
testcase_17 | AC | 224 ms
79,128 KB |
testcase_18 | AC | 244 ms
79,184 KB |
testcase_19 | AC | 120 ms
117,332 KB |
testcase_20 | AC | 122 ms
117,188 KB |
testcase_21 | AC | 123 ms
117,304 KB |
testcase_22 | AC | 123 ms
117,172 KB |
testcase_23 | AC | 121 ms
117,392 KB |
testcase_24 | AC | 71 ms
71,076 KB |
testcase_25 | AC | 74 ms
70,784 KB |
testcase_26 | AC | 72 ms
71,056 KB |
testcase_27 | AC | 71 ms
70,864 KB |
testcase_28 | AC | 72 ms
71,000 KB |
testcase_29 | AC | 72 ms
71,040 KB |
testcase_30 | AC | 593 ms
123,704 KB |
testcase_31 | AC | 615 ms
124,036 KB |
testcase_32 | AC | 470 ms
105,832 KB |
testcase_33 | AC | 671 ms
107,204 KB |
ソースコード
mod=998244353 def solv(n,k): ary0=[0]*(k+1) ary1=[0]*(k+1) ary0[1]=1 ary1[1]=(k-1)*n for i in range(2,k+1): # i以下のみで配列が作られている ary0[i]=pow(i,n,mod)-pow(i-1,n,mod)-n*pow(i-1,n-1,mod) # iより大がちょうど一つあり、それ以外はi以下で配列が作られている if i<k: # al:iより大がちょうど一つ # mi:iより大がちょうど一つ、降順で2番目がi未満->iは含まれない al=(k-i)*n # iより大の取り方と場所 al*=pow(i,n-1,mod) # i以下の数字 mi=(k-i)*n # iより大の取り方と場所 mi*=pow(i-1,n-1,mod) # i-1以下の数字 ary1[i]=al-mi # iより大がちょうど一つで、降順で2番目がi ans=0 for i in range(1,k+1): ans+=ary0[i]*i ans+=ary1[i]*i ans%=mod return ans n,k=map(int,input().split()) # コンビネーション。あらかじめO(N)の計算をすることでのちの計算が早くなる 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 g1=[1,1] # g1[i]=i! % mod :階乗 g2=[1,1] # g2[i]=(i!)^(-1) % mod :階乗の逆元 inverse=[0,1] for i in range(2,n+1): g1.append((g1[-1]*i)%mod) inverse.append((-inverse[mod%i]*(mod//i))%mod) g2.append((g2[-1]*inverse[-1])%mod) #print(solv_naive_ex(n,k)) #print(solv_naive(n,k)) print(solv(n,k))