結果
問題 | No.1897 Sum of 2nd Max |
ユーザー |
![]() |
提出日時 | 2022-04-08 23:02:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 519 ms / 2,000 ms |
コード長 | 1,412 bytes |
コンパイル時間 | 281 ms |
コンパイル使用メモリ | 82,440 KB |
実行使用メモリ | 59,136 KB |
最終ジャッジ日時 | 2024-11-28 14:05:25 |
合計ジャッジ時間 | 8,530 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
import sys #input = sys.stdin.readline #input = sys.stdin.buffer.readline #文字列はダメ #sys.setrecursionlimit(1000000) #import bisect #import itertools #import random #from heapq import heapify, heappop, heappush #from collections import defaultdict #from collections import deque #import copy #import math #from functools import lru_cache #@lru_cache(maxsize=None) #MOD = pow(10,9) + 7 MOD = 998244353 #dx = [1,0,-1,0] #dy = [0,1,0,-1] #dx8 = [1,1,0,-1,-1,-1,0,1] #dy8 = [0,1,1,1,0,-1,-1,-1] #dx = [1,1,-1,-1] #dy = [1,-1,1,-1] def main(): N,K = map(int,input().split()) ans = 0 for x in range(1,K+1): #print("X",x) val = 0 #最大が異なる場合 #最大の選び方がK-x通り、場所がN通り。 temp = N * (K - x) %MOD #残りのN-1箇所は少なくともxを一つ含む。 temp *= pow(x,N-1,MOD) - pow(x-1,N-1,MOD) val += temp val %= MOD #print(val) #最大が同じ場合(最大=x) #xを2個以上含む場合の数 zero = pow(x-1,N,MOD) #0個含む one = N * pow(x-1,N-1,MOD) #1個含む temp = pow(x,N,MOD) - (zero + one) temp %= MOD val += temp val %= MOD #print(zero,one,temp) #xの和 #print("VAL",val) ans += val * x %MOD ans %= MOD print(ans) if __name__ == '__main__': main()