完美正方形 深搜dfs

完美正方形

如果一些边长互不相同的正方形，可以恰好拼出一个更大的正方形，则称其为完美正方形。
历史上，人们花了很久才找到了若干完美正方形。比如：如下边长的22个正方形
2 3 4 6 7 8 12 13 14 15 16 17 18 21 22 23 24 26 27 28 50 60
如【图1.png】那样组合，就是一种解法。此时，
紧贴上边沿的是：60 50
紧贴下边沿的是：26 28 17 21 18
22阶完美正方形一共有8种。下面的组合是另一种：
2 5 9 11 16 17 19 21 22 24 26 30 31 33 35 36 41 46 47 50 52 61
如果告诉你该方案紧贴着上边沿的是从左到右依次为：47 46 61，
你能计算出紧贴着下边沿的是哪几个正方形吗？
请提交紧贴着下边沿的正方形的边长，从左到右，用空格分开。

思路：先将题目给的三个正方形填上，然后深搜，将剩下的19个正方形放置，直到找到解决方案。

代码：

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
int a[19]={2,5,9,11,16,17,19,21,22,24,26,30,31,33,35,36,41,50,52};
int mp[154][154];
bool vis[100];
void fill(int x,int y,int n,int num)//放置正方形  
{
	for(int i=x;i<x+n;i++)
		for(int j=y;j<y+n;j++)
			mp[i][j]=num;
}
bool judge(int x,int y,int n)//判断是否能放  
{
	if(x+n>154||y+n>154)
		return 0;
	for(int i=x;i<x+n;i++)
	{
		for(int j=y;j<y+n;j++)
		{
			if(mp[i][j])//要放的地方已经有正方形了  
				return 0;
		}
	}
	return 1;
}
bool slove()//判断是否全部放完  
{
	for(int i=0;i<154;i++)
	{
		for(int j=0;j<154;j++)
		{
			if(!mp[i][j])//还有空地，说明还没完全放完  
				return 0;
		}
	}
	return 1;
}
int dfs(int x,int y)
{
	if(slove())
		return 1;
	int flag=1;
	for(int i=0;i<154&&flag;i++)
	{
		for(int j=0;j<154&&flag;j++)
		{
			if(!mp[i][j])//找到一个没有放正方形的位置 
			{
				x=i;
				y=j;
				flag=0;
			}
		}
	}
	for(int i=0;i<19;i++)
	{
		if(judge(x,y,a[i]))
		{
			if(vis[i]==0)
			{
				fill(x,y,a[i],a[i]);
				vis[i]=1;
				if(dfs(x,y+a[i]))
					return 1;
				fill(x,y,a[i],0);//填0相当于取出正方形  
				vis[i]=0;
			}
		}
	}
	return 0;
}
int main()
{
	memset(mp,0,sizeof(mp));
	memset(vis,0,sizeof(vis));
	fill(0,0,47,47);//将题目给的三个正方形放上  
	fill(0,47,46,46);
	fill(0,93,61,61);
	dfs(0,0);
	int ans=0;
	for(int i=0;i<154;i++)
	{
		if(mp[153][i]!=ans)
		{
			cout<<mp[153][i]<<' ';//输出不同的四个数  
			ans=mp[153][i];
		}
	}
	return 0;
}