結果

問題 No.2503 Typical Path Counting Problem on a Grid
ユーザー chineristACchineristAC
提出日時 2023-10-13 23:57:09
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,939 ms / 2,000 ms
コード長 1,276 bytes
コンパイル時間 222 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 161,972 KB
最終ジャッジ日時 2024-09-15 19:50:18
合計ジャッジ時間 11,560 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge6
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 237 ms
140,928 KB
testcase_01 AC 519 ms
159,100 KB
testcase_02 AC 320 ms
156,160 KB
testcase_03 AC 952 ms
160,836 KB
testcase_04 AC 1,487 ms
161,972 KB
testcase_05 AC 773 ms
159,544 KB
testcase_06 AC 1,669 ms
161,000 KB
testcase_07 AC 1,939 ms
161,536 KB
testcase_08 AC 1,125 ms
160,328 KB
testcase_09 AC 1,406 ms
161,904 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from itertools import permutations
import heapq
from collections import deque
import random

input = lambda :sys.stdin.readline().rstrip()
mi = lambda :map(int,input().split())
li = lambda :list(mi())

mod = 998244353

def mat_mul(X,Y):
    n,m = len(X),len(Y[0])
    res = [[0 for j in range(m)] for i in range(n)]
    for i in range(n):
        for j in range(m):
            for k in range(len(Y)):
                res[i][j] += X[i][k] * Y[k][j]
                res[i][j] %= mod
    return res

M = 10**7 + 100
F = [0] * M
F[0] = 1
F[1] = 2
for n in range(2,M):
    F[n] = F[n-1] * (2 * n % mod) % mod  + F[n-2] * (n-1) % mod
    F[n] %= mod



def mat_pow(A,K):
    E = [[1,0],[0,1]]
    while K:
        if K & 1:
            E = mat_mul(E,A)
        A = mat_mul(A,A)
        K >>= 1
    return E

def solve(n,m):
    if m <= n:
        n,m = m,n
    
    if n == 0:
        return 1
    
    res = [[F[n]],[F[n-1]]]
    X = mat_pow([[2*n+1,n],[1,0]],m-n)
    res = mat_mul(X,res)

    


    if m == n:
        return (res[0][0] * F[n] % mod + (res[1][0] * n % mod) * F[n-1] % mod)  % mod
    else:
        return (res[0][0] * F[n] % mod + (res[1][0] * n % mod) * F[n-1] % mod)  % mod

for _ in range(int(input())):
    n,m = mi()
    print(solve(n,m))
0