結果

問題 No.1897 Sum of 2nd Max
ユーザー ygd.
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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()
0