結果
問題 | No.2676 A Tourist |
ユーザー |
![]() |
提出日時 | 2024-03-15 22:24:37 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 735 ms / 5,000 ms |
コード長 | 2,663 bytes |
コンパイル時間 | 4,295 ms |
コンパイル使用メモリ | 263,488 KB |
最終ジャッジ日時 | 2025-02-20 05:29:57 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 31 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint998244353; using namespace std; #define rep(i,n) for (int i = 0; i < (n); ++i) #define Inf32 1000000001 #define Inf64 1000000000000000001 struct HLD{ vector<int> sz,parent,depth,root,pos; vector<int> arr; HLD(vector<vector<int>> &E){ sz.resize(E.size(),1); parent.resize(E.size(),0); depth.resize(E.size(),0); root.resize(E.size(),0); pos.resize(E.size(),0); dfs(0,-1,E); dfs2(0,-1,E,0); } void dfs(int now,int p,vector<vector<int>> &E){ parent[now] = p; if(p==-1){ depth[now] = 0; } else{ depth[now] = depth[p]+1; } for(int i=0;i<E[now].size();i++){ int to = E[now][i]; if(to==p)continue; dfs(to,now,E); sz[now] += sz[to]; } } void dfs2(int now,int p,vector<vector<int>> &E,int r){ pos[now] = arr.size(); arr.push_back(now); root[now] = r; int maxi = 0; int ind = -1; for(int i=0;i<E[now].size();i++){ int to = E[now][i]; if(to==p)continue; if(maxi<sz[to]){ maxi = sz[to]; ind = to; } } if(ind==-1)return; dfs2(ind,now,E,r); for(int i=0;i<E[now].size();i++){ int to = E[now][i]; if(to==p||to==ind)continue; dfs2(to,now,E,to); } } vector<pair<int,int>> query(int u,int v){ vector<pair<int,int>> ret; int t = 0; while(root[u]!=root[v]){ if(depth[root[u]] <= depth[root[v]]){ ret.insert(ret.begin()+t,{pos[root[v]], pos[v]}); v = parent[root[v]]; } else{ ret.insert(ret.begin()+t,{pos[u],pos[root[u]]}); u = parent[root[u]]; t++; } } ret.insert(ret.begin()+t,{pos[u],pos[v]}); return ret; } int lca(int u,int v){ for(;;v=parent[root[v]]){ if(pos[u]>pos[v])swap(u,v); if(root[u]==root[v])return u; } } int get_distance(int u,int v){ return depth[u] + depth[v] - 2 * depth[lca(u,v)]; } }; int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); int n,q; cin>>n>>q; vector<long long> a(n); rep(i,n)cin>>a[i]; vector<vector<int>> E(n); rep(i,n-1){ int u,v; cin>>u>>v; u--,v--; E[u].push_back(v); E[v].push_back(u); } HLD H(E); fenwick_tree<long long> F(n); rep(i,n){ if(i!=0)F.add(H.pos[H.parent[i]],a[i]); } rep(_,q){ int t,x,y; cin>>t>>x>>y; if(t==0){ x--; if(x!=0){ F.add(H.pos[H.parent[x]],y); } a[x] += y; } else{ x--,y--; auto ret = H.query(x,y); long long ans = 0; rep(i,ret.size()){ int l = ret[i].first,r = ret[i].second; if(l>r)swap(l,r); ans += F.sum(l,r+1); } int l = H.lca(x,y); ans += a[l]; if(l!=0)ans += a[H.parent[l]]; cout<<ans<<endl; } } return 0; }