POJ 1321 经典棋盘问题的搜索和状态压缩解法

最新推荐文章于 2022-03-14 14:40:57 发布

Lei2015_

最新推荐文章于 2022-03-14 14:40:57 发布

阅读量750

点赞数

分类专栏： DP 搜索 ACM

本文链接：https://blog.csdn.net/u012935901/article/details/19927359

版权

ACM 同时被 3 个专栏收录

142 篇文章 0 订阅

订阅专栏

35 篇文章 0 订阅

订阅专栏

搜索

15 篇文章 0 订阅

订阅专栏

本文探讨的是简化版的n皇后问题，不考虑对角线限制。解决方案包括使用搜索算法，由于仅有8！种状态，可以直接计数。另外，通过状态压缩的动态规划方法，每行用二进制表示棋子位置，再结合转移操作，得出总的状态数为2^n * n * n。文章提供了两种不同的代码实现。

摘要由CSDN通过智能技术生成

这就是 n 皇后问题，不过简单了点，对于对角线不做限制

搜索，因为只有 8！个状态，直接数就可以了

状态压缩dp，每一行用一排1010代表上面有没有放过旗子，1是放过，0是没有，这一个行的状态可以是完全来自以前或者在这一行合法的地方放一个棋子。

每行有 2^n 种状态，共 n 行，每次转移要 n，总的就是 2^n*n*n;

两种代码

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>

using namespace std;
#define MAX 1000009 
#define INF 0x3f3f3f3f
#define MS(x) memset(x,0,sizeof(x))
#define ll long long
#define P pair<int,int>
#define fst first
#define sec second

int ans;
char b[20][20];
int n;
void dfs(int r,int state,int k)
{
	if(!k)
	{
		ans++;
		return ;
	}
	if(r>=n)
		return ;
	dfs(r+1,state,k);
	for(int i=0;i<n;i++)
	{
		if(((1<<i)&state)==0&&b[r][i]=='#')
		{
			dfs(r+1,(1<<i)|state,k-1);
		}
	}
}

int main()
{
	int k;
	while(scanf("%d%d",&n,&k)!=EOF&&n!=-1)
	{
		ans=0;
		for(int i=0;i<n;i++)
			scanf("%s",b[i]);
		dfs(0,0,k);
		cout<<ans<<endl;
	}
	return 0;
}

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>

using namespace std;
#define MAX 1000009 
#define INF 0x3f3f3f3f
#define MS(x) memset(x,0,sizeof(x))
#define ll long long
#define P pair<int,int>
#define fst first
#define sec second

int dp[10][1000];
char b[11][11];
int cnt[2000];
int n,k;
int check(int n)
{
	int ans=0;
	while(n)
	{
		if(n%2)
			ans++;
		n/=2;
	}
	return ans;
}
int main()
{
	for(int i=0;i<1000;i++)
		cnt[i]=check(i);
	while(scanf("%d%d",&n,&k)&&n!=-1)
	{
		MS(dp);
		for(int i=1;i<=n;i++)
			scanf("%s",b[i]+1);
		dp[0][0]=1;
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<(1<<n);j++)
			{
				if(cnt[j]<=k)
				{
					dp[i][j]+=dp[i-1][j];
					for(int p=1;p<=n;p++)
					{
						if( ((1<<p-1)&j) && b[i][p]=='#')
						{
							dp[i][j]+=dp[i-1][(~(1<<p-1))&j];
						}
					}
				}	
			}	
		}
		int ans=0;
		for(int i=0;i<(1<<n);i++)
			if(cnt[i]==k)
				ans+=dp[n][i];
		cout<<ans<<endl;
	}
	return 0;
}