poj1129Channel Allocation 图的染色问题

#include <iostream>

#include <cstdio>

#include <string>

using namespace std;

#define X 305

string s;

int map[X][X],color[X],n,x,y,len,c,ans;

bool check(int a,int col) //检查顶点a的颜色能否涂成col

{

for(int i=1;i<=n;i++)

if(map[a][i]&&color[i]==col) //相邻节点颜色与他一样，不行

return false;

return true;

}

bool dfs(int tc,int v) //搜索颜色总数为tc时，正在要涂色的是顶点v

{

if(tc==n) //如果颜色总数等于n，因为之前的用较少的颜色都不能涂好，所以此处可直接返回真

return true;

if(v==n+1) //如果把n个节点都涂完的话，直接返回真

return true;

for(int i=1;i<=tc;i++) //从颜色1开始枚举，看看当前颜色是否能涂满所有顶点

{

color[v] = i;

if(check(v,i)) //判断是否可以涂成颜色i

if(dfs(tc,v+1)) //如果下面的顶点同样能涂好，返回真

return true;

}

return false;

}

int solve() //根据平面图的4色定理，枚举颜色1到4即可

{

//////直接用1种颜色看看是否能涂好

bool flag = true;

for(int i=1;i<=n;i++)

for(int j=i+1;j<=n;j++)

if(map[i][j])

flag = false;

if(flag)

return 1;

for(int i=2;i<=4;i++) //从2开始枚举

{

memset(color,0,sizeof(color));

if(dfs(i,1))

return i;

}

return -1;

}

int main()

{

freopen("sum.in","r",stdin);

freopen("sum.out","w",stdout);

while(cin>>n,n)

{

ans = 1;

memset(map,0,sizeof(map));

for(int i=0;i<n;i++)

{

cin>>s;

len = s.size();

x = s[0] - 'A'+1; //转化为图的邻接矩阵表示形式，切下标从1开始

for(int j=2;j<len;j++)

{

c = s[j]-'A'+1;

map[x][c] = map[c][x] = 1;

}

}

int ans = solve();

if(ans==1)

printf("1 channel needed.\n");

else

printf("%d channels needed.\n",ans);

}

return 0;

}