CodeForces - 1244D 树（一条链）的染色

题意：给一个无向的无环的树，需要用三种颜色将他染色，相邻的三个点不能有重复的颜色。给出每个点染成每种颜色的花费，求最小的染色花费，如果给的图不能按要求染色，输出-1。

思路:只有三种颜色，相邻三个点还不能重复，说明一个点最多可以有两条边，那么这个图就是一条链。那首先就是用邻接表（也用过数组记录路径，但是不知道哪错了）把这条链用一个vector容器存起来，给这条链染色。因为只有三种颜色，所以只要确定了前两种，后面一条链的颜色就确定了。

注意：在第六组测试数据一直错，结果是因为main的初始值太小了，那就直接1e18 ，long long！！！

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<string.h>
#include<vector>
using namespace std;
vector<int>p;
struct node
{
    int cnt,c;
} qq[100010];
bool cmp(node a,node b)
{
    return a.cnt<b.cnt;
}
long long c[4][100010],sum;
int x[100010],y[100010],n,first[100010],next1[400110],book[100010];
void yao()
{
    int book[5];
    for(int j=3; j<n; j++)
    {
        book[1]=book[2]=book[3]=0;
        book[x[j-1]]=1;
        book[x[j-2]]=1;
        for(int i=1; i<=3; i++)
            if(!book[i])
            {
                qq[j].cnt=p[j];
                qq[j].c=i;
                x[j]=i;
                sum+=c[i][p[j]];
                break;
            }
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        memset(book,0,sizeof(book));
        p.clear();
        for(int i=0; i<=n; i++)
            first[i]=-1;
        for(int i=1; i<=3; i++)
            for(int j=1; j<=n; j++)
                scanf("%lld",&c[i][j]);
        int v[200010],u[200010],w=0,a,b;
        for(int i=1; i<=n-1; i++)
        {
            scanf("%d%d",&a,&b);
            book[a]++,book[b]++;
            u[w]=a,v[w]=b;
            next1[w]=first[u[w]];
            first[u[w]]=w;
            w++;
            u[w]=b,v[w]=a;
            next1[w]=first[u[w]];
            first[u[w]]=w;
            w++;
        }
        int f=0,t;
        for(int i=1; i<=n; i++)
            if(book[i]>2)
            {
                f=1;
                break;
            }
            else if(book[i]==1)
                t=i;
        if(f)
        {
            printf("-1\n");
            continue;
        }
        memset(book,0,sizeof(book));
        int j=first[t];
        p.push_back(u[j]);
        book[u[j]]=1;
        queue<int>q;
        q.push(t);
        while(!q.empty())
        {
            t=q.front();
            q.pop();
            for(int j=first[t]; j!=-1; j=next1[j])
            {
                if(book[v[j]])
                    continue;
                q.push(v[j]);
                book[v[j]]=1;
                p.push_back(v[j]);
            }
        }
        long long  minn=0x3f3f3f3f3f3f3f3f;
        int dir[6][3]= {1,2,3, 1,3,2, 2,1,3, 2,3,1, 3,2,1, 3,1,2};
        for(int i=0; i<6; i++)
        {
            sum=0;
            x[0]=dir[i][0];// 1 3 2
            x[1]=dir[i][1];
            x[2]=dir[i][2];
            qq[0].cnt=p[0],qq[0].c=x[0];
            qq[1].cnt=p[1],qq[1].c=x[1];
            qq[2].cnt=p[2],qq[2].c=x[2];
            sum=(c[x[0]][p[0]])+(c[x[1]][p[1]])+(c[x[2]][p[2]]);
            yao();
            if(sum<minn)
            {
                minn=sum;
                sort(qq,qq+n,cmp);
                for(int j=0; j<n; j++)
                    y[j]=qq[j].c;
            }
        }
        printf("%lld\n",minn);
        for(int i=0; i<n-1; i++)
            printf("%d ",y[i]);
        printf("%d\n",y[n-1]);
    }
    return 0;
}