CF593D Happy Tree Party

一、题目

点此看题

二、解法

可以树链剖分维护路径边权乘积，但是要卡精度。

一个数做除法的次数是$\log$级别的，但是要考虑$1$，观察题目条件，边权不会上升，等于一条边变成$1$之后就无法翻身了。可以用并查集加速暴力跳的过程，如果一条边边权为$1$就把他并到父亲那里去。

#include <cstdio>
#include <iostream>
using namespace std;
const int M = 200005;
#define int long long
int read()
{
	int x=0,f=1;char c;
	while((c=getchar())<'0' || c>'9') {if(c=='-') f=-1;}
	while(c>='0' && c<='9') {x=(x<<3)+(x<<1)+(c^48);c=getchar();}
	return x*f;
}
int n,m,tot,f[M],a[M],b[M],p[M],fa[M][20],dep[M],vx[M];
struct edge
{
	int v,c,next;
}e[2*M];
int find(int x)
{
	if(p[x]!=x) p[x]=find(p[x]);
	return p[x];
}
void dfs(int u,int pr)
{
	fa[u][0]=pr;
	dep[u]=dep[pr]+1;
	for(int i=1;i<=19;i++)
		fa[u][i]=fa[fa[u][i-1]][i-1];
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v,c=e[i].c;
		if(v==pr) continue;
		dfs(v,u);
		vx[v]=c;
		if(c==1) p[v]=find(u);
	}
}
int lca(int u,int v)
{
	if(dep[u]<dep[v]) swap(u,v);
	for(int i=19;i>=0;i--)
		if(dep[fa[u][i]]>=dep[v])
			u=fa[u][i];
	if(u==v) return u;
	for(int i=19;i>=0;i--)
		if(fa[u][i]^fa[v][i])
			u=fa[u][i],v=fa[v][i];
	return fa[u][0];
}
signed main()
{
	n=read();m=read();
	for(int i=1;i<n;i++)
	{
		int u=read(),v=read(),c=read();
		e[++tot]=edge{v,c,f[u]},f[u]=tot;
		e[++tot]=edge{u,c,f[v]},f[v]=tot;
		a[i]=u;b[i]=v;
	}
	for(int i=1;i<=n;i++) p[i]=i;
	dfs(1,0);
	while(m--)
	{
		int op=read(),u=read(),v=read();
		if(op==1)
		{
			int y=read(),t=lca(u,v);
			while(dep[u]>dep[t] && y)
			{
				y/=vx[u];
				u=find(fa[u][0]);
			}
			while(dep[v]>dep[t] && y)
			{
				y/=vx[v];
				v=find(fa[v][0]);
			}
			printf("%lld\n",y);
		}
		else
		{
			int x=dep[a[u]]>dep[b[u]]?a[u]:b[u];
			vx[x]=v;
			if(v==1) p[x]=find(fa[x][0]);
		}
	}
}