[IOI]2010 Race(dsu on tree)

題目

給出一棵樹，問長度爲K的簡單路徑，最少經過的邊數，假如沒有這種路徑則輸出-1，否則輸出邊數

題解

用map來存x節點下子樹上的每個節點距根節點的長度
這道題其實我也不太確定，交了十幾次，隨便改改小細節，答案都不一樣，唉，過題不怕不ac，就怕ac了還是不懂爲啥ac的

代碼

#include<bits/stdc++.h>
#include<tr1/unordered_map>
using namespace std;
#define ll long long
const int maxn=2e5+10;
ll head[maxn],sz[maxn],son[maxn],cnt,vis[maxn];
ll dis[maxn],ans,k,dep[maxn];
unordered_map<ll,ll> M;
struct node
{
	int to;
	ll w;
};
vector<node> g[maxn];
void dfs(int x,int f)
{
	//cout<<x<<' '<<f<<endl;
	sz[x]=1;dep[x]=dep[f]+1;
	for(auto i:g[x]){
		int v=i.to;
		//cout<<v<<endl;
		if(v!=f){
			dis[v]=dis[x]+i.w;
			dfs(v,x);
			sz[x]+=sz[v];
			if(!son[x]||sz[son[x]]<sz[v]) son[x]=v;
		}
	}
}
void cal(int x,int f,int r)
{
	//cout<<x<<' '<<f<<' '<<r<<endl;
	ll res=k+dis[r]*2-dis[x];
	if(M[res]) ans=min(ans,M[res]+dep[x]-2*dep[r]);
	for(auto i:g[x]){
		int v=i.to;
		if(v!=f&&!vis[v]){
			cal(v,x,r);
		}
	}
}
void update(int x,int f,int p)
{
	if(p>0){
		if(M[dis[x]]){
			M[dis[x]]=min(M[dis[x]],dep[x]);
		}
		else{
			M[dis[x]]=dep[x];
		}
	}
	else M[dis[x]]=0;
	for(auto i:g[x]){
		int v=i.to;
		if(v!=f&&!vis[v]) update(v,x,p);
	}
}
void dsu(int x,int f,int p){
	for(auto i:g[x]){
		int v=i.to;
		if(v!=f&&v!=son[x]){
			dsu(v,x,0);
		}
	}
	if(son[x]) dsu(son[x],x,1),vis[son[x]]=1;
	for(auto i:g[x]){
        int v=i.to;
        if(v!=f&&!vis[v]){
            update(v,x,1);
			cal(v,x,x);
        }
    }
    if(M[dis[x]]==0)
        M[dis[x]]=dep[x];
    else
        M[dis[x]]=min(M[dis[x]],dep[x]);
    if(M[dis[x]+k])
        ans=min(ans,M[dis[x]+k]-dep[x]);
	if(son[x]) vis[son[x]]=0;
	if(!p) update(x,f,-1);
}
int main()
{
	int n,x,y;
	ll z;
	scanf("%lld%lld",&n,&k);
	for(int i=1;i<=n-1;i++){
		scanf("%d %d %lld",&x,&y,&z);
		x++;y++;
		g[x].push_back(node{y,z});
		g[y].push_back(node{x,z});
	}
	dfs(1,0);
	ans=0x3f3f3f3f;
	dsu(1,0,0);
	if(ans==0x3f3f3f3f){
		printf("-1\n");
	}
	else{
		printf("%lld\n",ans);
	}
}