結果

問題 No.2963 Mecha DESU
コンテスト
ユーザー timi
提出日時 2024-11-20 11:35:51
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 572 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 614 ms
コンパイル使用メモリ 84,992 KB
実行使用メモリ 135,984 KB
最終ジャッジ日時 2026-05-14 14:09:39
合計ジャッジ時間 4,921 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge1_1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10 TLE * 1 -- * 46
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

N,M,K=map(int, input().split())
A=list(map(int, input().split()))


dp=[0]*(N+1)
mod=998244353
D={}
import heapq

for a in A:
  if a not in D:
    D[a]=0
  D[a]+=1
  
B=[]
for d in D:
  heapq.heappush(B,(d,d,D[d]))
	
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)
0