結果
問題 | 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) + 7MOD = 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 = 0for 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 += tempval %= 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 %= MODval += tempval %= MOD#print(zero,one,temp)#xの和#print("VAL",val)ans += val * x %MODans %= MODprint(ans)if __name__ == '__main__':main()