結果
問題 | No.2443 特殊線形群の標準表現 |
ユーザー | MasKoaTS |
提出日時 | 2023-08-10 12:18:01 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 3,930 bytes |
コンパイル時間 | 162 ms |
コンパイル使用メモリ | 81,836 KB |
実行使用メモリ | 171,728 KB |
最終ジャッジ日時 | 2024-04-28 05:35:47 |
合計ジャッジ時間 | 11,938 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 112 ms
86,044 KB |
testcase_01 | AC | 114 ms
86,064 KB |
testcase_02 | AC | 112 ms
85,888 KB |
testcase_03 | AC | 114 ms
86,016 KB |
testcase_04 | AC | 116 ms
85,888 KB |
testcase_05 | AC | 112 ms
85,836 KB |
testcase_06 | AC | 107 ms
86,016 KB |
testcase_07 | AC | 107 ms
85,760 KB |
testcase_08 | AC | 112 ms
85,760 KB |
testcase_09 | AC | 115 ms
85,888 KB |
testcase_10 | AC | 108 ms
86,144 KB |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | AC | 685 ms
164,992 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) """ SegmentTree from : https://github.com/customaddone/beginPython/blob/master/cgi-bin/library/segment_tree/segment_tree.py ref : https://algo-logic.info/segment-tree/ """ """ (単位元一覧) min:INF max:-INF XOR:0 区間和:0 区間積:1 最大公約数:0 """ ide_ele = [[1, 0], [0, 1]] #初期値(単位元) """ update(k,x) :k番目の要素をxに更新する query(l,r) : [l, r)(l <= k < r の区間)から値を取得する """ def segfunc(B, A): #求めたい処理(maxやmin、XORなど) ret = [[0, 0], [0, 0]] for i in range(2): for j in range(2): for k in range(2): ret[i][j] += A[i][k] * B[k][j] ret[i][j] %= b return ret class SegmentTree: """ init(init_val, ide_ele): 配列init_valで初期化 O(N) update(k, x): k番目の値をxに更新 O(logN) query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN) """ def __init__(self, init_val, segfunc, ide_ele): """ init_val: 配列の初期値 segfunc: 区間にしたい操作 ide_ele: 単位元 n: 要素数 num: n以上の最小の2のべき乗 tree: セグメント木(1-index) """ n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1 << (n - 1).bit_length() self.tree = [ide_ele] * 2 * self.num # 配列の値を葉にセット for i in range(n): self.tree[self.num + i] = init_val[i] # 構築していく for i in range(self.num - 1, 0, -1): self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1]) def update(self, k, x): """ k番目の値をxに更新 k: index(0-index) x: update value """ k += self.num self.tree[k] = x while k > 1: self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1]) k >>= 1 def query(self, l, r): """ [l, r)のsegfuncしたものを得る l: index(0-index) r: index(0-index) """ res = self.ide_ele l += self.num r += self.num prev = 0 while l < r: if l & 1: res = self.segfunc(res, self.tree[l]) if(l > prev) else self.segfunc(self.tree[l], res) prev = l l += 1 if r & 1: res = self.segfunc(res, self.tree[r - 1]) if(l > prev) else self.segfunc(self.tree[r - 1], res) prev = r l >>= 1 r >>= 1 prev >>= 1 return res """ Main Code """ n, b, q = getNs() As = [[[x % b for x in getNs()] for _ in [0] * 2] for _ in [0] * n] query = [[x % b for x in getNs()] for _ in [0] * q] # print(As) # print(query) seg = SegmentTree(As, segfunc, ide_ele) def prod_vector(A, vec): ret = [0, 0] for i in range(2): for j in range(2): ret[i] += A[i][j] * vec[j] ret[i] %= b return ret for l, r, x, y in query: vec = [x, y] # print(seg.query(l, r)) ans = prod_vector(seg.query(l, r), vec) print(*ans)