Codeforces Round #617 (Div. 3) - F . Berland Beauty (1296F)

题意：给定 n 个点的树并给出 n-1 条边，每条边有一定的权值(但我们事先不知道)，给出 m 个游览记录，每个记录给出 u,v,w 代表顶点 u 和 v 之间的路径上边的权值最小为 w ;问是否能构造出这样一棵树(即与给定的n-1条边是否冲突)，能则按给出边的顺序输出边的权值，不能则输出 -1

 

分析：因为 n<=5000 ，所以我们可以把 m 组游览记录按 w 从大到小排序，并分别遍历它的路径上的所有边，如果所有边在之前都已经标记过了，则证明无解 (因为如果所有路径在之前都标记了，则证明这条路径上最小的权值就是之前被标记过的那些里最小的产生矛盾)

 

代码：

#include<cstdio>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;

typedef pair<int,int> P;
const int   N = 5000+5;
const int INF = 1E6;

struct node{
	int u,v,w;
	node(){}
	node(int _u,int _v,int _w){
		u=_u,v=_v,w=_w;
	}
}p[N];
bool cmp(node a,node b){
	return a.w>b.w;
}

vector<P>e[N];
int n,m;
int  fa[N];
int dep[N];
int cnt[N];
int num[N];
int ans[N];

void dfs(int u,int f)
{
	fa[u]=f;
	dep[u]=dep[f]+1;
	for(int i=0;i<e[u].size();i++){
		if(e[u][i].first==f) continue;
		dfs(e[u][i].first,u);
		cnt[e[u][i].first]=e[u][i].second;
	}
}

bool check(int u,int v,int w)
{
	int minn=INF;
	if(dep[u]<dep[v]) swap(u,v);
	while(dep[u]!=dep[v]){
		if(num[u]==INF) num[u]=w;
		minn=min(minn,num[u]);
		u=fa[u]; 
	}
	while(u!=v){
		if(num[u]==INF) num[u]=w;
		minn=min(minn,num[u]);
		u=fa[u];
		if(num[v]==INF) num[v]=w;
		minn=min(minn,num[v]);
		v=fa[v];
	}
	return minn==w;
}

int main()
{
	scanf("%d",&n);
	for(int i=1,u,v;i<n;i++){
	    scanf("%d%d",&u,&v);
	    e[u].push_back(P(v,i));
	    e[v].push_back(P(u,i));
	}
	dfs(1,0);
	for(int i=1;i<=n;i++) num[i]=INF;
	
	scanf("%d",&m);
	for(int i=0,u,v,w;i<m;i++){
		scanf("%d%d%d",&u,&v,&w);
		p[i]=node(u,v,w);
	}
	sort(p,p+m,cmp);
	for(int i=0;i<m;i++){
		if(!check(p[i].u,p[i].v,p[i].w)){
			puts("-1"); return 0;
		}
	}
	for(int i=2;i<=n;i++) ans[cnt[i]]=num[i];
	for(int i=1;i<n;i++) printf("%d ",ans[i]);
}