結果
問題 | No.2503 Typical Path Counting Problem on a Grid |
ユーザー |
|
提出日時 | 2023-10-13 23:56:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,724 ms / 2,000 ms |
コード長 | 1,216 bytes |
コンパイル時間 | 289 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 162,048 KB |
最終ジャッジ日時 | 2024-09-15 19:49:07 |
合計ジャッジ時間 | 11,255 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 10 |
ソースコード
import sysfrom itertools import permutationsimport heapqfrom collections import dequeimport randominput = lambda :sys.stdin.readline().rstrip()mi = lambda :map(int,input().split())li = lambda :list(mi())mod = 998244353def 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] %= modreturn resM = 10**7 + 100F = [0] * MF[0] = 1F[1] = 2for n in range(2,M):F[n] = F[n-1] * 2 * n + F[n-2] * (n-1)F[n] %= moddef 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 >>= 1return Edef solve(n,m):if m <= n:n,m = m,nif n == 0:return 1res = [[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] + res[1][0] * n * F[n-1]) % modelse:return (res[0][0] * F[n] + res[1][0] * n * F[n-1]) % modfor _ in range(int(input())):n,m = mi()print(solve(n,m))