题目
给出一棵树,问长度为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);
}
}