結果

問題 No.1836 Max Matrix
ユーザー tamatotamato
提出日時 2022-02-11 22:29:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,213 ms / 2,000 ms
コード長 1,005 bytes
コンパイル時間 372 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 60,544 KB
最終ジャッジ日時 2024-06-27 19:57:08
合計ジャッジ時間 9,795 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 998244353
eps = 10**-9


def main():
    import sys
    input = sys.stdin.readline

    H, W, M = map(int, input().split())

    ans = 0
    for m in range(1, M + 1):
        tmp = pow(M - m + 1, H, mod) - pow(M - m, H, mod)
        if m == 1:
            tmp2 = pow(M, W, mod) - pow(M - 1, W, mod)
        else:
            tmp2 = pow(M - m + 1, W, mod) - pow(M - m, W, mod)
        ans = (ans + (tmp * tmp2)%mod)%mod

        tmp = pow(M - m + 1, W, mod) - pow(M - m, W, mod)
        if m == 1:
            tmp2 = pow(M, H, mod) - pow(M - 1, H, mod)
        else:
            tmp2 = pow(M - m + 1, H, mod) - pow(M - m, H, mod)
        ans = (ans + (tmp * tmp2)%mod) % mod

        # same
        tmp = pow(M - m + 1, H, mod) - pow(M - m, H, mod)
        if m == 1:
            tmp2 = pow(M, W, mod) - pow(M - 1, W, mod)
        else:
            tmp2 = pow(M - m + 1, W, mod) - pow(M - m, W, mod)
        ans = (ans - (tmp * tmp2)%mod) % mod
    print(ans)


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