DFS练习-codevs-1116四色问题

1116 四色问题

时间限制: 1 s
空间限制: 128000 KB
题目等级 : 黄金 Gold

题目描述 Description

给定N（小于等于8）个点的地图，以及地图上各点的相邻关系，请输出用4种颜色将地图涂色的所有方案数（要求相邻两点不能涂成相同的颜色）

数据中0代表不相邻，1代表相邻
输入描述 Input Description

第一行一个整数n，代表地图上有n个点

接下来n行，每行n个整数，每个整数是0或者1。第i行第j列的值代表了第i个点和第j个点之间是相邻的还是不相邻，相邻就是1，不相邻就是0.

我们保证a[i][j] = a[j][i] （a[i,j] = a[j,i]）
输出描述 Output Description

染色的方案数
样例输入 Sample Input

8
0 0 0 1 0 0 1 0
0 0 0 0 0 1 0 1
0 0 0 0 0 0 1 0
1 0 0 0 0 0 0 0
0 0 0 0 0 0 0 0
0 1 0 0 0 0 0 0
1 0 1 0 0 0 0 0
0 1 0 0 0 0 0 0
样例输出 Sample Output

15552
数据范围及提示 Data Size & Hint

n<=8
题解

#include <stdio.h>
#include <string.h>

int n;
int sum=0;
int contact[9][9];
int s[9];

void dfs(int num)
{
    int i,flag;
    bool color[5];
    memset(color,0,sizeof(color));
    if(num==n)
    {
        flag=0;
        for(i=1;i<=n;i++)
            if(contact[num][i])
            {
                color[s[i]]=1;
            }
        for(i=1;i<5;i++)
            if(!color[i])
            {
                flag++;
            }
        if(flag)
            sum+=flag;
        return;
    }
    if(num==1)
    {
        memset(s,0,sizeof(s));
        for(i=1;i<=4;i++)
        {
            s[1]=i;
            dfs(2);
        }
        return;
    }
    else
    {
        for(i=1;i<=n;i++)
        {
            if(contact[num][i])
            {
                color[s[i]]=1;
            }
        }
        for(i=1;i<=4;i++)
        {
            if(!color[i])
            {
                s[num]=i;
                dfs(num+1);
                s[num]=0;
            }
        }
        return;
    }

    return;
}

int main()
{
    int i,j;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        for(j=1;j<=n;j++)
        {
            scanf("%d",&contact[i][j]);
        }
    }
    dfs(1);
    printf("%d\n",sum);
    return 0;
}