提出 d7c8d1b6-74a7-426e-85dd-3b28db6b5568

コード

import math
import bisect
import collections
import heapq
#import sys
#import decimal
#import random

mod = 998244353
def inverse(x):
    ret = 1
    y = mod - 2
    while y > 0:
        if cppMod(y, 2) == 1:
            ret = cppMod(ret * x, mod)
        x = cppMod(x * x, mod)
        y >>= 1
    return ret

def cppMod(x,y):
    if (x >= 0 and y >= 0) or (x <= 0 and y <= 0):
        return x % y
    else:
        return 0 if x % y == 0 else x-y*(x//y+1)

def main():
    n,m,k = map(int,input().split())
    vec = list(map(int,input().split()))
    ans = 0
    pm = 1
    combi = 1
    for i in range(k+1):
        num = 0
        if (m - 1 - i >= 0):
            num = vec[m - 1 - i]
        ans += pm * combi * num
        ans = cppMod((cppMod(ans, mod) + mod), mod)
        pm *= -1
        combi = cppMod((combi * (k-1)), mod)
        combi = cppMod((combi * inverse(i + 1)), mod)
    print(ans)
    return

main()

結果

問題	点数	言語	結果	実行時間	メモリ
O. Decode	0	Python3 (CPython)	TLE	3012 ms	28980 KiB

テストセット

テストセット	結果	得点	実行時間	メモリ

詳細結果

テストケース	結果	実行時間	メモリ
subtask1_example_01.txt	WA	30 ms	9856 KiB
subtask1_hand_01.txt	AC	30 ms	9984 KiB
subtask1_hand_02.txt	WA	28 ms	9984 KiB
subtask1_hand_03.txt	AC	28 ms	9728 KiB
subtask1_random_01.txt	WA	29 ms	10496 KiB
subtask1_random_02.txt	WA	48 ms	17408 KiB
subtask1_random_03.txt	WA	100 ms	28980 KiB
subtask2_example_01.txt	WA	30 ms	9856 KiB
subtask2_hand_01.txt	TLE	2923 ms	9856 KiB
subtask2_hand_02.txt	TLE	3012 ms	8192 KiB
subtask2_hand_03.txt	TLE	2939 ms	12800 KiB
subtask2_hand_04.txt	TLE	3007 ms	8448 KiB
subtask2_random_01.txt	WA	1309 ms	11008 KiB
subtask2_random_02.txt	WA	1922 ms	16872 KiB
subtask2_random_03.txt	TLE	2519 ms	15532 KiB
subtask2_random_04.txt	WA	1487 ms	10240 KiB
subtask2_random_05.txt	TLE	2649 ms	15536 KiB
subtask2_random_06.txt	TLE	2786 ms	23328 KiB
subtask2_random_07.txt	TLE	2901 ms	21764 KiB
subtask2_random_08.txt	TLE	2680 ms	25376 KiB