【CODE[VS]】1116--四色问题

题目描述 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 <iostream>
using namespace std;

int n, sum;
int step[10];
bool vis[10][10];

bool check(int n){
    int i, j;
    for(i = 0; i < n; ++i){
        for(j = 0; j < n; ++j){
            if(vis[i][j] && step[i] == step[j]){
                return false;
            }
        }
    }
    return true;
}

void dfs(int d){
    int i;
    if(d == n){
        ++sum;
        return;
    }
    for(i = 1; i <= 4; ++i){
        step[d] = i;
        if(check(d + 1)){
            dfs(d + 1);
        }
    }
}

int main(){
    int i, j;
    cin >> n;
    for(i = 0; i < n; ++i){
        for(j = 0; j < n; ++j){
            cin >> vis[i][j];
        }
    }
    dfs(0);
    cout << sum << endl;
    return 0;
}