codeforces1296F Berland Beauty

https://codeforces.com/problemset/problem/1296/F

这场div3也太水了= =

只要记录每一条边最小限制中最大的是多少，他就应该取多少，然后看这个取值满足了多少个询问的答案。

如果所有询问的答案都满足完了，那么就是可行解，否则是不可行的。

用vector mle on test268，开数组过了。

#include<bits/stdc++.h>
using namespace std;

const int maxl=5e3+10;

int n,m,cnt;
int fa[maxl],fae[maxl],ehead[maxl],dep[maxl],ans[maxl];
struct ed
{
	int to,nxt,id;
}e[maxl<<1];
typedef pair<int,int> p;
int len[maxl];
p ee[maxl][maxl];
bool in[maxl];

inline void add(int u,int v,int id)
{
	e[++cnt].to=v;e[cnt].id=id;
	e[cnt].nxt=ehead[u];ehead[u]=cnt;
}

inline void dfs(int u,int f)
{
	int v;
	for(int i=ehead[u];i;i=e[i].nxt)
	{
		v=e[i].to;
		if(v==f) continue;
		dep[v]=dep[u]+1;
		fa[v]=u;fae[v]=e[i].id;
		dfs(v,u);
	}
}

inline void prework()
{
	scanf("%d",&n);
	int u,v,mi;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v,i);add(v,u,i);
	}
	dfs(1,0);
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&u,&v,&mi);
		if(dep[u]<dep[v])
			swap(u,v);
		while(dep[u]>dep[v])
		{	
			//ee[fae[u]].push_back(make_pair(mi,i));
			ee[fae[u]][++len[fae[u]]]=make_pair(mi,i);
			u=fa[u];
		}
		while(u!=v)
		{
			//ee[fae[u]].push_back(make_pair(mi,i));
			ee[fae[u]][++len[fae[u]]]=make_pair(mi,i);
			//ee[fae[v]].push_back(make_pair(mi,i));
			ee[fae[v]][++len[fae[v]]]=make_pair(mi,i);
			u=fa[u];v=fa[v];
		}
	}
}

inline bool cmp(const p &a,const p &b)
{
	if(a.first==b.first)
		return a.second<b.second;
	return a.first>b.first;
}

inline void mainwork()
{
	int l;
	for(int i=1;i<n;i++)
	{	
		if(!len[i]) 
		{
			ans[i]=1;
			continue;
		}
		sort(ee[i]+1,ee[i]+len[i]+1,cmp);
		ans[i]=ee[i][1].first;
		for(int j=1;j<=len[i];j++)
		if(ee[i][j].first==ans[i])
			in[ee[i][j].second]=true;
	}
}

inline void print()
{
	int flag=1;
	for(int i=1;i<=m;i++)
	if(!in[i])
	{
		puts("-1");
		return;
	}
	for(int i=1;i<n;i++)
		printf("%d%c",ans[i],(i==n-1)?'\n':' ');
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}