【NOIP2017提高A组模拟9.23】Carry

题目

题目描述

数据范围

题解

题目大意

有一棵树，每次要从$x$走到$y$，权值为走过的最大边的长度。可以使得一条边的长度减少$l$，求最小总权值

题目分析

$50\%$

可以暴力枚举修改哪条边的长度，用倍增维护修改后的每次查询，时间复杂度$O(nlog(n))$

$100\%$

发现只有修改路径上的最大值才是可能影响答案的，而且修改后新的值变成了次大值和最大值$-l$之间更大的那个，那么我们就给最大值打上一个标记，记录说如果更改这条边会影响多少，最后又未修改的答案减去影响最大的边的值，就是最终答案

Code

#include<cstdio>
#include<iostream>
using namespace std;
struct node
{
	int to,val,head,next;
};
node a[200005];
int n,q,l,x,y,z,mn,tot,ans;
int deep[100005],f[100005][20],mx[100005][20],mx2[100005][20],id[100005][20],sum[100005];
int read()
{
	int res=0,fh=1;char ch=getchar();
	while (ch<'0'||ch>'9') {if (ch=='-') fh=-1;ch=getchar();}
	while (ch>='0'&&ch<='9') res=res*10+(ch-'0'),ch=getchar();
	return res*fh;
}
void add(int x,int y,int z)
{
	a[++tot].to=y;
	a[tot].val=z;
	a[tot].next=a[x].head;
	a[x].head=tot;
}
void dfs(int now,int fa,int x)
{
	deep[now]=deep[fa]+1;
	f[now][0]=fa;
	mx[now][0]=x;
	id[now][0]=now;
	for (int i=1;i<=19;++i)
	{
		f[now][i]=f[f[now][i-1]][i-1];
		if (mx[now][i-1]>mx[f[now][i-1]][i-1])
		{
			mx[now][i]=mx[now][i-1];
			id[now][i]=id[now][i-1];
			mx2[now][i]=max(mx2[now][i-1],mx[f[now][i-1]][i-1]);
		}
		else
		{
			mx[now][i]=mx[f[now][i-1]][i-1];
			id[now][i]=id[f[now][i-1]][i-1];
			mx2[now][i]=max(mx2[f[now][i-1]][i-1],mx[now][i-1]);
		}
	}
	for (int i=a[now].head;i;i=a[i].next)
	{
		if (a[i].to==fa) continue;
		dfs(a[i].to,now,a[i].val);
	}
}
void count(int x,int i,int& MX,int& MX2,int& ID)
{
	if (mx[x][i]>MX)
	{
		MX2=max(MX2,max(MX,mx2[x][i]));
		MX=mx[x][i];
		ID=id[x][i];
	}
	else MX2=max(mx[x][i],MX2);
}
void flag(int mx,int mx2,int id)
{
	ans+=mx;
	if (mx2==mx) return;
	sum[id]+=max(mx2-mx,-l);
}
void solve(int x,int y)
{
	int mx=-2147483647,mx2=-2147483647,id=0;
	if (x==y) return;
	if (deep[x]<deep[y]) swap(x,y);
	if (deep[x]>deep[y])
		for (int i=19;i>=0;--i)
			if (deep[f[x][i]]>=deep[y])
			{
				count(x,i,mx,mx2,id);
				x=f[x][i];
			}
	if (x==y)
	{
		flag(mx,mx2,id);
		return;
	}
	for (int i=19;i>=0;--i)
	{
		if (f[x][i]!=f[y][i])
		{
			count(x,i,mx,mx2,id);
			count(y,i,mx,mx2,id);
			x=f[x][i],y=f[y][i];
		}
	}
	count(x,0,mx,mx2,id);
	count(y,0,mx,mx2,id);
	flag(mx,mx2,id);
}
int main()
{
	freopen("carry.in","r",stdin);
	freopen("carry.out","w",stdout);
	n=read();q=read();l=read();
	for (int i=1;i<n;++i)
	{
		x=read();y=read();z=read();
		add(x,y,z);add(y,x,z);
	}
	dfs(1,0,0);
	for (int i=1;i<=q;++i)
	{
		x=read();y=read();
		solve(x,y);
	}
	int mn=2147483647;
	for (int i=1;i<=n;++i)
		mn=min(mn,sum[i]);
	printf("%d\n",ans+mn);
	fclose(stdin);
	fclose(stdout);
	return 0;
}