CodeForces 593D Happy Tree Party [LCA+并查集]

 1 #include<bits/stdc++.h>
 2 #define fi first
 3 #define se second
 4 using namespace std;
 5 typedef long long ll;
 6 const int inf=1e9;
 7 const double PI=acos(-1);
 8 const double eps=1e-6;
 9 const int mod=1e9+7;
10 const int maxn=2e5+10;
11 int n,m;
12 typedef pair<int,int> pii;
13 vector<pii>f[maxn];
14 ll val[maxn];
15 int pos[maxn],pre[maxn],dep[maxn],g[maxn];
16 int find(int k){
17     if(k==pre[k]) return k;
18     return pre[k]=find(pre[k]);//并查集
19 }
20 void dfs(int u,int fa){
21     g[u]=fa;//u的爸爸
22     for(pii x:f[u]){
23         if(x.fi==fa) continue;
24         pos[x.fi]=x.se;//记录每个点对应的边权的编号，边权转点权
25         dep[x.fi]=dep[u]+1;//计算深度
26         dfs(x.fi,u);
27     }
28 }
29 void up(int u){
30     pre[u]=g[u];//u变成u的爸爸
31 }
32 ll lca(int a,int b,ll y){
33     a=find(a),b=find(b);
34     while(a!=b){
35         if(dep[a]<dep[b]) swap(a,b);//每次将较深的点向上找
36         if(val[pos[a]]==1) up(a);//点权为1时，压缩该点
37         y/=val[pos[a]];a=find(g[a]);//除以该点的权值
38         if(y==0) return 0;//y为0直接跳出
39     }
40     return y;
41 }
42 int main(){
43     ios::sync_with_stdio(false);
44     //freopen("in","r",stdin);
45     cin>>n>>m;
46     for(int i=0;i<=n;i++) pre[i]=i;
47     for(int i=1,u,v;i<n;i++){
48         cin>>u>>v>>val[i];
49         f[u].push_back(pii(v,i));
50         f[v].push_back(pii(u,i));
51     }
52     dfs(1,0);
53     for(int i=1;i<=m;i++){
54         int op,a,b,p;
55         ll c,y;
56         cin>>op;
57         if(op==1){
58             cin>>a>>b>>y;
59             cout<<lca(a,b,y)<<endl;
60         }else{
61             cin>>p>>c;
62             val[p]=c;
63         }
64     }
65     return 0;
66 }