【练习07】 DFS 1024 N皇后问题

最新推荐文章于 2021-02-23 10:16:35 发布

liuzhushiqiang

最新推荐文章于 2021-02-23 10:16:35 发布

阅读量587

点赞数

分类专栏：【练习07】 DFS 文章标签：图论 DFS

本文链接：https://blog.csdn.net/liuzhushiqiang/article/details/9532641

版权

【练习07】 DFS 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

算法思路：DFS。

这一题可以用DFS + 回溯的方法进行暴力枚举。但还是要注意可行性的判断，及时的break以减少时间开销。

要注意的是虽然n最大只有10，但是case会很多，如果每次都DFS会TLE，所以要打表。这个问题只能怪测试数据太无聊了。

自己刚开始做的时候没有注意vis数组下标最大会达到 2 * n - 1，导致数组开得太小，访问了非法内存，WA了，可怕的是像这种情况系统并不会有任何提示，照样运行，还好及时发现，下次吸取教训。

//模板开始
#include <string>   
#include <vector>   
#include <algorithm>   
#include <iostream>   
#include <sstream>   
#include <fstream>   
#include <map>   
#include <set>   
#include <cstdio>   
#include <cmath>   
#include <cstdlib>   
#include <ctime>
#include<iomanip>
#include<string.h>
#define SZ(x) (int(x.size()))
using namespace std;

int toInt(string s){
	istringstream sin(s); 
	int t; 
	sin>>t; 
	return t;
}
template<class T> string toString(T x){
	ostringstream sout; 
	sout<<x; 
	return sout.str();
}
typedef long long int64;
int64 toInt64(string s){
	istringstream sin(s); 
	int64 t; 
	sin>>t;
	return t;
}
template<class T> T gcd(T a, T b){ 
	if(a<0) 
		return gcd(-a, b);
	if(b<0) 
		return gcd(a, -b);
	return (b == 0)? a : gcd(b, a % b);
}
//模板结束（通用部分）

#define ifs cin

int n;
#define MAXN 25
int C[MAXN];
int tot;
int vis[3][MAXN];
int dabiao[MAXN];


void search(int cur)
{
	int i, j;
	if(cur == n)
	{
		tot++;
	}
	else
	{
		for(i = 0; i < n; i++)
		{
			if(!vis[0][i] && !vis[1][cur + i] && !vis[2][cur - i + n])
			{
				C[cur] = i;
				vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 1;
				search(cur + 1);
				vis[0][i] = vis[1][cur + i] = vis[2][cur - i + n] = 0;
			}
		}
	}
}

//【练习07】 DFS 1024 N皇后问题

int main()
{
	//ifstream ifs("shuju.txt", ios::in);
	for(n = 1; n <= 12; n++)		//打表。
	{
		tot = 0;
		memset(vis, 0, sizeof(vis));
		search(0);
		dabiao[n] = tot;
	}
	while(ifs>>n && n != 0)
	{
		cout<<dabiao[n]<<endl;
	}

	return 0;
}