蓝桥杯——完美正方形（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，
你能计算出紧贴着下边沿的是哪几个正方形吗？

请提交紧贴着下边沿的正方形的边长，从左到右，用空格分开。

不要填写任何多余的内容或说明文字。

答案：50 33 30 41

参考博客

思路：

这里就是把这个大正方形想象成一个大方格，然后往里边填数

#include<cstring>
#include<iostream>
#include<algorithm>
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 m[154][154];
int vis[100];

void fill(int x,int y,int n,int num){//把数据写进m 
    for(int i=x; i<x+n; i++)
        for(int j=y; j<y+n; j++)
            m[i][j]=num;
}

bool solve(){//是不是m已经填满 
    for(int i=0; i<154; i++)
        for(int j=0; j<154; j++)
            if(!m[i][j])
                return false;
    return true;
}

bool judge(int x,int y,int n){//判断是否可以写 
    if(x+n>154||y+n>154)
        return false;
    for(int i=x; i<n+x; i++)
        for(int j=y; j<n+y; j++)
            if(m[i][j])
                return false;
    return true;
}

double dfs(int x,int y){
    if(solve())
        return true;
    else{
        int flag=1;
        for(int i=0; i<154&&flag; i++)
            for(int j=0; j<154&&flag; j++)
                if(!m[i][j]){
                    x=i;
					y=j;
					flag=0;              	
				}
        for(int k=0; k<19; k++){
            if(judge(x,y,a[k])) {
                if(!vis[k]){
                    fill(x,y,a[k],a[k]);
                    vis[k]=1;
                    if(dfs(x,y+a[k]))
                        return true;
                    fill(x,y,a[k],0);
                    vis[k]=0;
                }
            }
            else
                break;
        }
    }
    return false;
}

int main(){
    memset(m,0,sizeof(m));
    memset(vis,0,sizeof(vis));
    fill(0,0,47,47);
    fill(0,47,46,46);
    fill(0,93,61,61);//预先把给出的三个正方形写进m 
    dfs(0,0);
    int ans=0;
    for(int i=0; i<154; i++) {//打印最底下的边 
        if(m[153][i]!=ans){
            printf("%d ",m[153][i]);
            ans=m[153][i];
        }
    }
    return 0;
}