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

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

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

#include <cstdio>
#include <cstring>
#include <string>
#include <bits/stdc++.h>
#include <cmath>
#include <queue>
#include <set>
#include <map>
#include <vector>
#include <algorithm>
#define ll long long

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 G[154][154];
int vis[100];

void fill(int x,int y,int size,int num){
    for (int i=x;i<x+size;i++){
        for (int j=y;j<y+size;j++){
            G[i][j]=num;
        }
    }
}

bool judge(int x,int y,int size){
    if (x+size>154||y+size>154){
        return false;
    }
    for (int i=x;i<x+size;i++){
        for (int j=y;j<y+size;j++){
            if (G[i][j]!=0){
                return false;
            }
        }
    }
    return true;
}

bool solve(){
    for (int i=0;i<154;i++){
        for (int j=0;j<154;j++){
            if (G[i][j]==0){
                return false;
            }
        }
    }
    return true;
}

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

    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(G[153][i]!=ans)
        {
            printf("%d ",G[153][i]);
            ans=G[153][i];
        }
    }
}