思路很简单:
读入每一条确定边权的路径,然后求这个路径两点间LCA路线,然后把一路上没有给边权的赋值这个确定了的边权,如果有边权,并且边权大于等于该确定的边权,那么继续,否则不符合条件
代码参考别人的,我写的时候WA了很多发
code:
#include<bits/stdc++.h>
using namespace std;
const int N = 5e3+5;
struct node{
int u;
int i;
};
vector<node>v[N];
int dis[N],d[N],f[N],num[N],va[N],need[N];
void dfs(int u,int fa)//lca前dfs
{
d[u]=d[fa]+1;
f[u]=fa;
for(int i=0;i<v[u].size();i++)
{
int k=v[u][i].u;
if(k==fa) continue;
dfs(k,u);
num[k]=v[u][i].i;记录终点为k的边
}
va[u]=1e6;
}
int LCA(int x,int y,int w)
{
int minn = 1e6;
if(d[x]<d[y]) swap(x,y);
while(d[x]!=d[y]){
if(va[x]==1e6) va[x] = w;
minn = min(minn,va[x]);
x = f[x];
}
while(x!=y){
if(va[x]==1e6) va[x] = w;
minn = min(minn,va[x]); x = f[x];
if(va[y]==1e6) va[y] = w;
minn = min(minn,va[y]);y = f[y];
}
return minn==w;
}
struct so
{
int a,b,w;
bool operator < (const so A) const
{return w<A.w;}
};
priority_queue<so>q;//必须从大到小,反之,第二组数据就会直接无解,因为第二组数据可能LCA沿途的边的边权可能大于第一组。
int main()
{
int n,m;
cin>>n;
for(int i=1;i<n;i++)
{
int a,b;
cin>>a>>b;
v[a].push_back({b,i});
v[b].push_back({a,i});
}
dfs(1,0);
cin>>m;
for(int i=1;i<=m;i++)
{
int u,v,w;
cin>>u>>v>>w;
q.push({u,v,w});
}
while( !q.empty()){
so we = q.top();
q.pop();
if(!LCA(we.a,we.b,we.w))
{
puts("-1");
return 0;
}
}
for(int i=2;i<=n;i++)
need[num[i]]=va[i];
for(int i=1;i<n;i++)
cout<<need[i]<<" ";
}