深度优先搜索与回溯法

例题分析

数独是一种著名的益智游戏。这里讨论的是一种简化过的数独--四阶数独。给出一个4*4的格子，每个格子只能填写1到4的整数，要求每行、每列和四等分更小的正方形部分都刚好由1到4组成。

给出空白的方格，请问：一共有多少种合法的填写方法？

解决这个问题最朴素的枚举方式是，一共有16个选项，每个空格可以填1到4，所以考虑使用16层for循环，一共有大约是42亿种情况。显然这种方式非常低效，不可行。

这里介绍一种新的函数递归枚举的方式，枚举每一个填空中所有可能的选项，然后判断这种选项是否合法。如果这个选项合法的话就填写下一个选项，然后继续；如果这个填空中所有的选项都不合法，那就不用继续枚举下去了，而是去尝试更换上一个填空的选项，继续枚举。这种方式称为回溯算法，尝使用深度优先搜索来实现。

回溯算法的一般形式如下：

void dfs(int k){//k代表递归层数，或者说要填第几个空
   if(所有空已经填完了){
       判断最优解/记录答案
       return;
   }
   for(枚举这个空能填的选项){
      if(这个选项是合法的){
          记录下这个空(保存现场);
          dfs(k+1);
          取消这个空(恢复现场);
      }
    }
}

例题代码 

#include<iostream>
using namespace std;
#define size 5
int a[size*size],n=4*4,ans=0;
int b1[size][5],b2[size][5],b3[size][5];//分别记录横行、竖行、四小块
void dfs(int x){
	if(x>n){
		ans++;
		return;
	}
	int row=(x-1)/4+1;//横行编号
	int col=(x-1)%4+1;//竖排编号
	int block=(row-1)/2*2+(col-1)/2+1;//小块编号
	for(int i=1;i<=4;i++){
		if(b1[row][i]==0&&b2[col][i]==0&&b3[block][i]==0){
			a[x]=i;//记录放置位置
			b1[row][i]=1;b2[col][i]=1;b3[block][i]=1;//占位置
			dfs(x+1);
			b1[row][i]=0;b2[col][i]=0;b3[block][i]=0;//取消占位 
		}
	} 
}
int main(){
	dfs(1);
	cout<<ans;
	return 0;
}