結果

問題 No.2550 MORE! JUMP! MORE!
コンテスト
ユーザー Seed57_cash
提出日時 2023-11-21 17:43:10
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,203 bytes
コンパイル時間 306 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 100,224 KB
最終ジャッジ日時 2024-09-26 07:15:57
合計ジャッジ時間 5,091 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20 TLE * 1 -- * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 998244353


def solve(n, a_list):
    # k段目にm回目のジャンプで到達する方法 = (k - 1)C(m - 1)
    # そのあとの方法 = 2 ** (n - k - 1)
    # k段目が起因の得点の合計 = a_k * sum_{m = 1}^{k} m * (k - 1)C(m - 1)
    # = a_k * (2 ** (k - 1) + sum_{m = 1}^{k} (m - 1) * (k - 1)C(m - 1))
    # = a_k * (2 ** (k - 1) + (k - 1) * 2 ** (k - 2))
    if n == 1:
        return a_list[0]
    res = 0
    for i in range(1, n + 1):
        if 2 <= i < n:
            res += a_list[i - 1] * (pow(2, i - 1, MOD) + (i - 1) * pow(2, i - 2, MOD)) * pow(2, n - i - 1)
        elif i == 1:
            res += a_list[i - 1] * (pow(2, i - 1, MOD) + (i - 1)) * pow(2, n - i - 1)
        elif i == n:
            res += a_list[i - 1] * (pow(2, i - 1, MOD) + (i - 1) * pow(2, i - 2, MOD))
        res %= MOD
    # print(res)
    return res


def main():
    n = int(input())
    a_list = list(map(int, input().split()))
    res = solve(n, a_list)
    print(res)


def test():
    assert solve(3, [5, 7, 8]) == 95
    assert solve(7, [
        376873723, 252623151, 856139513, 29843394, 373100489, 753241875, 92750716
    ]) == 686211551


if __name__ == "__main__":
    test()
    main()
0