ACM 1116 四色问题(DFS)

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

很简单的DFS

#include <iostream>
#include <cstring>
#include <vector>

using namespace std;

#define MAX_N 8
int n;
int G[MAX_N][MAX_N];
int cnt=0;
bool vis[MAX_N];
int color[MAX_N];

bool OK()
{
    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    {
        if(G[i][j] && color[i]==color[j]) return false;
    }

    return true;
}

void dfs(int cur,int cr)
{
    color[cur]=cr;
    if(cur==n-1)
    {
        if(OK())
        {
            cnt++;
        }
        return;
    }

    for(int i=0;i<4;i++)
        dfs(cur+1,i);
}

int main()
{
    cin>>n;

    for(int i=0;i<n;i++)
    for(int j=0;j<n;j++)
    {
        cin>>G[i][j];
    }

    memset(vis,0,sizeof(vis));

    for(int i=0;i<4;i++)
        dfs(0,i);

    cout<<cnt<<endl;

    return 0;
}