結果

問題 No.2229 Treasure Searching Rod (Hard)
ユーザー LyricalMaestro
提出日時 2025-01-03 03:44:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 995 ms / 2,000 ms
コード長 914 bytes
コンパイル時間 442 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 105,536 KB
最終ジャッジ日時 2025-01-03 03:45:09
合計ジャッジ時間 22,285 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

## https://yukicoder.me/problems/no/2229

MOD = 998244353

def main():
    H, W, K = map(int, input().split())
    xyv = []
    for _ in range(K):
        x, y, v = map(int, input().split())
        xyv.append((x, y, v))

    answer = 0
    for x, y, v in xyv:
        ans = x

        ans0 = ((x - 1) * x) % MOD
        ans0 *= pow(2, MOD -2, MOD)
        ans0 %= MOD

        d = max(0, x - y)
        ans1 = (d * (d + 1)) % MOD
        ans1 *= pow(2, MOD - 2, MOD)
        ans1 %= MOD
        ans += (ans0 - ans1) % MOD
        ans %= MOD

        
        d = max(0, x - (W + 1 - y))
        ans2 = (d * (d + 1)) % MOD
        ans2 *= pow(2, MOD - 2, MOD)
        ans2 %= MOD
        ans += (ans0 - ans2) % MOD
        ans %= MOD

        answer += (v * ans) % MOD
        answer %= MOD
    print(answer)

    



                


                



    


        




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