Berland Beauty——codeforce1296F(LCA+构造)

思路很简单:
读入每一条确定边权的路径,然后求这个路径两点间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]<<" ";
	
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值