CODEVS 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

分类标签 Tags 点此展开 

#include <iostream>
#include <algorithm>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <vector>
using namespace std;
vector<int> v[10];
int a[10];
int n;
long long cnt;

void dfs(int cur){
	if(cur==n+1){
		cnt++;
		return;
	}
	for(int i=1;i<=4;i++){
		int flag=0;
		for(int j=0;j<v[cur].size();j++){
			if(a[v[cur][j]]==i){
				flag=1;
				break; 
			} 
		}
		if(flag)
			continue;	
		a[cur]=i;
		dfs(cur+1);
		a[cur]=0;
	}
}

int main(){
	
	scanf("%d",&n);
	memset(a,0,sizeof(a));
	for(int i=0;i<=n;i++)
		v[i].clear();
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			int x;
			scanf("%d",&x);
			if(x){
				v[i].push_back(j);
			}
		}
	}
	dfs(1);
	printf("%lld\n",cnt);
	return 0;
}