結果
問題 | No.2443 特殊線形群の標準表現 |
ユーザー | MasKoaTS |
提出日時 | 2023-08-09 23:16:15 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 5,636 bytes |
コンパイル時間 | 298 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 283,988 KB |
最終ジャッジ日時 | 2024-11-15 21:33:24 |
合計ジャッジ時間 | 14,709 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 137 ms
85,504 KB |
testcase_01 | AC | 134 ms
85,376 KB |
testcase_02 | AC | 132 ms
85,632 KB |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | RE | - |
testcase_07 | RE | - |
testcase_08 | RE | - |
testcase_09 | AC | 133 ms
85,376 KB |
testcase_10 | AC | 131 ms
85,248 KB |
testcase_11 | RE | - |
testcase_12 | RE | - |
testcase_13 | RE | - |
testcase_14 | RE | - |
testcase_15 | RE | - |
testcase_16 | RE | - |
testcase_17 | RE | - |
testcase_18 | RE | - |
testcase_19 | RE | - |
testcase_20 | AC | 1,937 ms
283,988 KB |
ソースコード
import itertools as iter import collections as coll import heapq as hq import bisect as bis from decimal import Decimal as dec from functools import cmp_to_key import math import sys #import pypyjit #pypyjit.set_param('max_unroll_recursion=-1') sys.setrecursionlimit(10 ** 6) inp = sys.stdin.readline input = lambda : inp()[:-1] getN = lambda : int(inp()) getNs = lambda : map(int, inp().split()) getList = lambda :list(map(int, inp().split())) getStrs = lambda n : [input() for _ in [0] * n] def yexit(): print("Yes"); exit(0) def nexit(): print("No"); exit(0) pi = 3.141592653589793 mod = 1000000007 MOD = 998244353 INF = 4611686018427387903 dx = [1, 0, -1, 0]; dy = [0, 1, 0, -1] #di = coll.defaultdict(int) class ModInt: mod = 998244353 def __init__(self, x): self.x = x % ModInt.mod @classmethod def set_mod(cls, mod: int) -> None: ModInt.mod = mod def __str__(self): return str(self.x) __repr__ = __str__ def __add__(self, other): return (ModInt(self.x + other.x) if isinstance(other, ModInt) else ModInt(self.x + other)) def __sub__(self, other): return (ModInt(self.x - other.x) if isinstance(other, ModInt) else ModInt(self.x - other)) def __mul__(self, other): return (ModInt(self.x * other.x) if isinstance(other, ModInt) else ModInt(self.x * other)) def __truediv__(self, other): return (ModInt(self.x * self.inv_mod(other.x, self.mod)) if isinstance(other, ModInt) else ModInt(self.x * self.inv_mod(other, self.mod))) def __pow__(self, other): return ModInt(pow(self.x, other.x, ModInt.mod)) if isinstance(other, ModInt) else ModInt(pow(self.x, other, ModInt.mod)) __radd__ = __add__ def __rsub__(self, other): return (ModInt(other.x - self.x) if isinstance(other, ModInt) else ModInt(other - self.x)) __rmul__ = __mul__ def __rtruediv__(self, other): return (ModInt(other.x * self.inv_mod(self.x, self.mod)) if isinstance(other, ModInt) else ModInt(other * self.inv_mod(self.x, self.mod))) def __rpow__(self, other): return ModInt(pow(other.x, self.x, ModInt.mod)) if isinstance(other, ModInt) else ModInt(pow(other, self.x, ModInt.mod)) def __iadd__(self, other): self = self + other return self def __isub__(self, other): self = self - other return self def __imul__(self, other): self = self * other return self def __itruediv__(self, other): self = self / other return self @classmethod def inv_gcd(cls, n, m): n %= m if(n == 0): return m, 0 s, t, m0, m1 = m, n, 0, 1 while(t): u = s // t s -= t * u m0 -= m1 * u m0, m1, s, t = m1, m0, t, s if(m0 < 0): m0 += m // s return s, m0 @classmethod def inv_mod(cls, n, m): _, im = cls.inv_gcd(n, m) return im @classmethod def prod(cls, A): ret = ModInt(1) for k in A: ret *= k return ret @classmethod def nPk(cls, n, k): if(isinstance(n, ModInt)): n = n.x if(isinstance(k, ModInt)): k = k.x return cls.prod([*range(n - k + 1, n + 1)]) @classmethod def nCk(cls, n, k): if(isinstance(n, ModInt)): n = n.x if(isinstance(k, ModInt)): k = k.x r = min(n - k, k) return cls.nPk(n, r) / cls.nPk(r, r) class Matrix: def __init__(self, A: list): if not(A and isinstance(A[0], list)): A = [A] self.A = A self.row = len(A) self.col = len(A[0]) self.I = None if(self.row == self.col): self.I = self.calc_I() def set_A(self, A: list) -> None: self.A = A def calc_I(self) -> list: ret = [[int(i == j) for j in range(self.col)] for i in range(self.row)] return ret def __str__(self): return str(self.A) __repr__ = __str__ def __add__(self, other): ret = Matrix(self) for i in range(self.row): for j in range(self.col): ret.A[i][j] = self.A[i][j] + other.A[i][j] return ret def __sub__(self, other): ret = Matrix(self) for i in range(self.row): for j in range(self.col): ret.A[i][j] = self.A[i][j] - other.A[i][j] return ret def __mul__(self, other): assert(self.col == other.row) ret = Matrix([[0]*other.col for _ in [0]*self.row]) for i in range(self.row): for j in range(other.col): ret.A[i][j] = sum(self.A[i][k] * other.A[k][j] for k in range(self.col)) return ret def __pow__(self, other: int): mat = self ret = Matrix(self.I) n = other while(n): if(n & 1): ret = mat * ret mat = mat * mat n >>= 1 return ret """ Main Code """ n, b, q = getNs() ModInt.set_mod(b) A_lis = [[[ModInt(x) for x in getNs()] for _ in [0] * 2] for _ in [0] * n] query = [getList() for _ in [0] * q] As = [Matrix(A) for A in A_lis] A_prod = Matrix(As[0].calc_I()) Bs = [A_prod] for A in As: A_prod = A * A_prod Bs.append(A_prod) # print(Bs) def inverse(A): return Matrix([[A.A[1][1], -A.A[0][1]], [-A.A[1][0], A.A[0][0]]]) for l, r, x, y in query: vec = Matrix([[ModInt(x)], [ModInt(y)]]) ans = Bs[r] * inverse(Bs[max(0, l - 1)]) * vec print(ans.A[0][0], ans.A[1][0])