結果
問題 |
No.650 行列木クエリ
|
ユーザー |
![]() |
提出日時 | 2018-02-28 12:30:54 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 769 bytes |
コンパイル時間 | 196 ms |
コンパイル使用メモリ | 82,152 KB |
実行使用メモリ | 67,456 KB |
最終ジャッジ日時 | 2024-12-17 19:23:35 |
合計ジャッジ時間 | 2,433 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | RE * 1 |
other | RE * 10 |
ソースコード
import numpy as np n = int(input()) graph = [[] for _ in range(n)] for i in range(n-1): a, b = map(int, input().split()) graph[a].append((i,b)) graph[b].append((i,a)) parents = [0]*n edges = [0]*(n-1) def dfs(u, p): parents[u] = p for i, v in graph[u]: if v == p: continue edges[i] = v dfs(v, u) dfs(0, 0) mats = [np.array([[1,0],[0,1]])]*n for _ in range(int(input())): a = input().split() if a[0] == 'x': i, a, b, c, d = map(int, a[1:]) x = np.array([[a,b],[c,d]]) mats[edges[i]] = x else: i, j = map(int, a[1:]) c = mats[j] while i != parents[j]: j = parents[j] c = np.dot(mats[j], c) print(*(e for r in c for e in r))