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]);
}