POJ 1129 Channel Allocation(dfs)

Description
给出n个点的邻接表,问最小点染色
Input
第一行为一整数n表示点数,之后n行每行输入一个点的邻接表,以n=0结束输入(1<=n<=26)
Output
输出这张图的最小点染色
Sample Input
2
A:
B:
4
A:BC
B:ACD
C:ABD
D:BC
4
A:BCD
B:ACD
C:ABD
D:ABC
0
Sample Output
1 channel needed.
3 channels needed.
4 channels needed.
Solution
简单dfs,dfs(num,now),now表示当前要要染的颜色,num表示当前已经使用的颜色数,用color[i]记录每个点的颜色,那么从0~num-1枚举now点的颜色,如果满足条件(即now的邻接点中没有颜色为i的点)则dfs(num,now+1),枚举完后令color[now]=num,然后dfs(num+1,now+1),注意回溯,注意输出时数量为1则channel没有s
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
#define maxn 33
#define INF 0x3f3f3f3f
struct node
{
    int to,next;
}edge[maxn*maxn];
int head[maxn],color[maxn],tot,n,ans;
char s[maxn];
void init()
{
    memset(head,-1,sizeof(head));
    memset(color,-1,sizeof(color));
    tot=0;ans=INF;
}
void add(int u,int v)
{
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}
bool check(int u)
{
    for(int i=head[u];~i;i=edge[i].next)
    {
        int v=edge[i].to;
        if(color[u]==color[v]) return 0;
    }
    return 1;
}
void dfs(int num,int now)
{
    if(now==n)
    {
        ans=min(ans,num);
        return ;
    }
    for(int i=0;i<num;i++)
    {
        color[now]=i;
        if(check(now))dfs(num,now+1);
    }
    color[now]=num;
    dfs(num+1,now+1);
    color[now]=-1;
}
int main()
{
    while(scanf("%d",&n),n)
    {
        init();
        for(int i=0;i<n;i++)
        {
            scanf("%s",s);
            int len=strlen(s);
            for(int j=2;j<len;j++)
                add(i,s[j]-'A'),add(s[j]-'A',i);
        }
        dfs(0,0);
        if(ans==1)printf("1 channel needed.\n");
        else printf("%d channels needed.\n",ans);
    }
    return 0;
}

你可能感兴趣的:(POJ 1129 Channel Allocation(dfs))