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

题意：给一棵树，每条边有一个权值，给两种操作，第一种是询问y向下整除从a到b的最短路径中每条边的权值后y的值，第二种是改变某条边的权值。

思路：y的最大值为1e18，最多除大于等于2的数不超过60次即可将y变为0，先dfs以任意一点为根建树，记录每个点的深度和它的父结点并将边权转化为点权，

再搞个并查集，将权值为1的点压缩，即使pre[u]=g[u];(u变成u的爸爸)。

 #include<bits/stdc++.h>
 #define fi first
 #define se second
 using namespace std;
 typedef long long ll;
 const int inf=1e9;
 const double PI=acos(-);
 const double eps=1e-;
 const int mod=1e9+;
 const int maxn=2e5+;
 int n,m;
 typedef pair<int,int> pii;
 vector<pii>f[maxn];
 ll val[maxn];
 int pos[maxn],pre[maxn],dep[maxn],g[maxn];
 int find(int k){
     if(k==pre[k]) return k;
     return pre[k]=find(pre[k]);//并查集
 }
 void dfs(int u,int fa){
     g[u]=fa;//u的爸爸
     for(pii x:f[u]){
         if(x.fi==fa) continue;
         pos[x.fi]=x.se;//记录每个点对应的边权的编号，边权转点权
         dep[x.fi]=dep[u]+;//计算深度
         dfs(x.fi,u);
     }
 }
 void up(int u){
     pre[u]=g[u];//u变成u的爸爸
 }
 ll lca(int a,int b,ll y){
     a=find(a),b=find(b);
     while(a!=b){
         if(dep[a]<dep[b]) swap(a,b);//每次将较深的点向上找
         if(val[pos[a]]==) up(a);//点权为1时，压缩该点
         y/=val[pos[a]];a=find(g[a]);//除以该点的权值
         if(y==) return ;//y为0直接跳出
     }
     return y;
 }
 int main(){
     ios::sync_with_stdio(false);
     //freopen("in","r",stdin);
     cin>>n>>m;
     for(int i=;i<=n;i++) pre[i]=i;
     for(int i=,u,v;i<n;i++){
         cin>>u>>v>>val[i];
         f[u].push_back(pii(v,i));
         f[v].push_back(pii(u,i));
     }
     dfs(,);
     for(int i=;i<=m;i++){
         int op,a,b,p;
         ll c,y;
         cin>>op;
         if(op==){
             cin>>a>>b>>y;
             cout<<lca(a,b,y)<<endl;
         }else{
             cin>>p>>c;
             val[p]=c;
         }
     }
     return ;
 }