Codeforces Round #592 (Div. 2) D. Paint the Tree（构造）

题目链接


题意：给定一棵树，要求你给每个节点涂色，要求连续的三个节点的颜色不能相同，每个节点涂三个颜色的花费分别是a【i】，b【i】，c【i】，问你满足要求的最小花费。
思路：很显然，画个图就知道能满足条件的只有链的情况，于是我们只要固定从根节点开始的前三个节点的颜色，那么整条链的颜色就都确定了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
vector<int>g[maxn];
ll a[maxn],b[maxn],c[maxn],ans[7][maxn],minn=2e18,temp;
void dfs(int x,int fa,int a,int b,int c,int deep,int id)
{
	if(deep%3==1) ans[id][x]=a;
	else if(deep%3==2) ans[id][x]=b;
	else if(deep%3==0) ans[id][x]=c;
	for(int to:g[x])
	{
		if(to==fa) continue;
		dfs(to,x,a,b,c,deep+1,id);
	}
}
int main()
{
	int k,n,x,y,root=1;
   scanf("%d",&n);
   for(int i=1;i<=n;++i) scanf("%lld",&a[i]);
   for(int i=1;i<=n;++i) scanf("%lld",&b[i]);
   for(int i=1;i<=n;++i) scanf("%lld",&c[i]);
   for(int i=1;i<n;++i)
   {
   		scanf("%d %d",&x,&y);
   		g[x].push_back(y);
   		g[y].push_back(x);
   }
   for(int i=1;i<=n;++i) 
   {
   	if(g[i].size()==1) root=i;
   	if(g[i].size()>2){
   	puts("-1");return 0;
   }
   }
   dfs(root,0,1,2,3,1,1);dfs(root,0,1,3,2,1,2);dfs(root,0,2,1,3,1,3);dfs(root,0,2,3,1,1,4);dfs(root,0,3,2,1,1,5);dfs(root,0,3,1,2,1,6);
   for(int i=1;i<=6;++i)
   {
   	temp=0;
   	for(int j=1;j<=n;++j)
   	{
   		if(ans[i][j]==1) temp+=a[j];
   		else if(ans[i][j]==2) temp+=b[j];
   		else temp+=c[j];
	   }
	   if(temp<minn) minn=temp,k=i;
   }
   printf("%lld\n",minn);
   for(int i=1;i<=n;++i) printf("%d ",ans[k][i]);
}