提出 9aaa28c9-df7a-4702-82ad-dc4b0c32ce88

コード

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-i)), mod)
        combi = cppMod((combi * inverse(i + 1)), mod)
    print(ans)
    return

main()

結果

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

テストセット

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

詳細結果

テストケース	結果	実行時間	メモリ
subtask1_example_01.txt	AC	29 ms	9856 KiB
subtask1_hand_01.txt	AC	30 ms	9856 KiB
subtask1_hand_02.txt	AC	31 ms	9856 KiB
subtask1_hand_03.txt	AC	31 ms	9856 KiB
subtask1_random_01.txt	AC	31 ms	10496 KiB
subtask1_random_02.txt	AC	53 ms	17476 KiB
subtask1_random_03.txt	AC	76 ms	29060 KiB
subtask2_example_01.txt	AC	31 ms	9984 KiB
subtask2_hand_01.txt	TLE	2923 ms	9856 KiB
subtask2_hand_02.txt	TLE	3008 ms	8192 KiB
subtask2_hand_03.txt	TLE	2959 ms	12800 KiB
subtask2_hand_04.txt	TLE	3001 ms	20704 KiB
subtask2_random_01.txt	AC	1280 ms	10880 KiB
subtask2_random_02.txt	TLE	2279 ms	16876 KiB
subtask2_random_03.txt	TLE	2556 ms	15412 KiB
subtask2_random_04.txt	AC	1508 ms	10368 KiB
subtask2_random_05.txt	TLE	2648 ms	15844 KiB
subtask2_random_06.txt	TLE	2804 ms	23128 KiB
subtask2_random_07.txt	TLE	2904 ms	21944 KiB
subtask2_random_08.txt	TLE	2774 ms	25356 KiB