BZOJ 1087 SCOI 2005 상호 불가침 킹상압 DP

제목 대의: n*n의 국제 장기 바둑판을 정하고 그 위에 k개의 국왕을 놓아 국왕 간에 서로 공격하지 않기를 요구하며 방안의 수를 구한다.
n<=⑨
상압 DP는 각 줄의 방안을 2진법으로 1차원으로 압축하여 f[i][j][k]를 i행으로 하고 j개 국왕의 상태가 k인 방안수를 사용한 다음에 상태를 다음과 같이 옮긴다.
f[i][j][k]=Σf[i-1][j-digit[k]][l]
그중 l&k=0, l>1&k=0, l<<1&k=0, digit[k]는 k의 2진법 중 1의 개수
폭력으로 옮기면 돼요.
롱롱 꼭 켜주세요.
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
int n,m;
bool map[512][512],usable[512];
ll f[10][100][512],ans;
int digit[512];
bool Judge(int x,int y)
{
	if(x&y)
		return false;
	if(x<<1&y)
		return false;
	if(x>>1&y)
		return false;
	return true;
}
bool Usable(int x)
{
	if(x<<1&x)
		return false;
	if(x>>1&x)
		return false;
	return true;
}
int Get_Digit(int x)
{
	int re=0;
	while(x)
		re+=x&1,x>>=1;
	return re;
}
int main()
{
	int i,j,k,l;
	cin>>n>>m;
	for(i=0;i<1<<n;i++)
		for(j=0;j<1<<n;j++)
			if( Judge(i,j) )
				map[i][j]=1;
	for(i=0;i<1<<n;i++)
		digit[i]=Get_Digit(i);
	for(i=0;i<1<<n;i++)
		usable[i]=Usable(i);
	f[0][0][0]=1;
	for(i=1;i<=n;i++)
		for(j=0;j<=m;j++)
			for(k=0;k<1<<n;k++)
				if(usable[k]&&digit[k]<=j)
					for(l=0;l<1<<n;l++)
						if(usable[l]&&map[k][l]&&digit[k]+digit[l]<=j)
							f[i][j][k]+=f[i-1][j-digit[k]][l];
	for(i=0;i<1<<n;i++)
		ans+=f[n][m][i];
	cout<<ans<<endl;
}

좋은 웹페이지 즐겨찾기