結果
問題 | No.2640 traO Stamps |
ユーザー |
![]() |
提出日時 | 2024-02-19 22:16:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 802 ms / 2,000 ms |
コード長 | 2,949 bytes |
コンパイル時間 | 388 ms |
コンパイル使用メモリ | 82,092 KB |
実行使用メモリ | 105,248 KB |
最終ジャッジ日時 | 2024-09-29 02:12:14 |
合計ジャッジ時間 | 21,949 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 33 |
ソースコード
N,M,K=map(int, input().split())S=list(map(int, input().split()))def warshall_floyd(d):#d[i][j]: iからjへの最短距離for k in range(n):for i in range(n):for j in range(n):d[i][j] = min(d[i][j],d[i][k] + d[k][j])return d##############################n,w =N,Md = [[float("inf")]*n for i in range(n)]#d[u][v] : 辺uvのコスト(存在しないときはinf)for i in range(w):x,y,z = map(int,input().split())x-=1;y-=1d[x][y] = z;d[y][x] = zfor i in range(n):d[i][i] = 0 #自身のところに行くコストは0B=(warshall_floyd(d))#####segfunc#####def segfunc(x, y):return x+y######################ide_ele#####ide_ele = 0#################class SegTree:"""init(init_val, ide_ele): 配列init_valで初期化 O(N)update(k, x): k番目の値をxに更新 O(N)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 = segfuncself.ide_ele = ide_eleself.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.numself.tree[k] = xwhile k > 1:self.tree[k >> 1] = self.segfunc(self.tree[k], self.tree[k ^ 1])k >>= 1def query(self, l, r):"""[l, r)のsegfuncしたものを得るl: index(0-index)r: index(0-index)"""res = self.ide_elel += self.numr += self.numwhile l < r:if l & 1:res = self.segfunc(res, self.tree[l])l += 1if r & 1:res = self.segfunc(res, self.tree[r - 1])l >>= 1r >>= 1return resA=[]for i in range(K):p,q=S[i],S[i+1]A.append(B[p-1][q-1])seg = SegTree(A, segfunc, ide_ele)Q=int(input())for i in range(Q):t,x,y=map(int, input().split())if t==1:p,q=x-1,xif x-1>=0:l=S[x-1]-1d=B[l][y-1]seg.update(x-1,d)if x+1<=K:r=S[x+1]-1d=B[r][y-1]seg.update(x,d)S[x]=yelse:print(seg.query(x,y))# print(seg.query(0, 8))# seg.update(5, 0)# print(seg.query(0, 8))