結果

問題 No.1897 Sum of 2nd Max
ユーザー ygd.ygd.
提出日時 2022-04-08 23:02:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 468 ms / 2,000 ms
コード長 1,412 bytes
コンパイル時間 403 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 59,136 KB
最終ジャッジ日時 2024-05-06 08:45:05
合計ジャッジ時間 8,175 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
51,840 KB
testcase_01 AC 35 ms
51,584 KB
testcase_02 AC 35 ms
52,096 KB
testcase_03 AC 390 ms
58,752 KB
testcase_04 AC 430 ms
58,496 KB
testcase_05 AC 221 ms
58,880 KB
testcase_06 AC 219 ms
58,496 KB
testcase_07 AC 264 ms
58,496 KB
testcase_08 AC 222 ms
58,752 KB
testcase_09 AC 395 ms
58,624 KB
testcase_10 AC 361 ms
58,496 KB
testcase_11 AC 383 ms
58,624 KB
testcase_12 AC 383 ms
58,752 KB
testcase_13 AC 388 ms
58,880 KB
testcase_14 AC 125 ms
58,496 KB
testcase_15 AC 128 ms
58,624 KB
testcase_16 AC 177 ms
59,136 KB
testcase_17 AC 153 ms
59,136 KB
testcase_18 AC 164 ms
58,752 KB
testcase_19 AC 36 ms
51,840 KB
testcase_20 AC 35 ms
51,456 KB
testcase_21 AC 35 ms
52,096 KB
testcase_22 AC 36 ms
52,352 KB
testcase_23 AC 34 ms
52,004 KB
testcase_24 AC 35 ms
51,840 KB
testcase_25 AC 37 ms
52,096 KB
testcase_26 AC 35 ms
51,456 KB
testcase_27 AC 35 ms
51,456 KB
testcase_28 AC 34 ms
52,416 KB
testcase_29 AC 33 ms
51,456 KB
testcase_30 AC 398 ms
58,496 KB
testcase_31 AC 415 ms
58,496 KB
testcase_32 AC 335 ms
58,624 KB
testcase_33 AC 468 ms
59,136 KB
権限があれば一括ダウンロードができます

ソースコード

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