結果

問題 No.2963 Mecha DESU
ユーザー timi
提出日時 2024-11-20 11:26:05
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 565 bytes
コンパイル時間 619 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 184,832 KB
最終ジャッジ日時 2024-11-20 11:27:31
合計ジャッジ時間 78,053 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 23 TLE * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

N,M,K=map(int, input().split())
A=list(map(int, input().split()))
dp=[0]*(N+1)
mod=998244353
D={}
for a in A:
  if a not in D:
    D[a]=0
  D[a]+=1
  
B=[]
for d in D:
  B.append((d,d,D[d]))
    
import heapq
for i in range(1,N+1):
    while B:
        x,y,z=heapq.heappop(B)
        if i!=x:
            heapq.heappush(B,(x,y,z))
            break 
        else:
            dp[x]+=z
            dp[x]%=mod
            x+=y
            heapq.heappush(B,(x,y,z))
    
D={};ans=0;pp=pow(M,K,mod)
for i in range(1,N+1):
  d=dp[i]
  if d in D:
    ans+=D[d]
  else:
    p=pow(M-d,K,mod)
    ans+=pp-p
    D[d]=pp-p
print(ans*pow(pp,-1,mod)%mod)
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0