結果

問題 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)
0