2015年蓝桥杯 完美正方形

题目：

如果一些边长互不相同的正方形，可以恰好拼出一个更大的正方形，则称其为完美正方形。
历史上，人们花了很久才找到了若干完美正方形。
比如：如下边长的22个正方形
2 3 4 6 7 8 12 13 14 15 16 17 18 21 22 23 24 26 27 28 50 60
如下图那样组合，就是一种解法。此时，
紧贴上边沿的是：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
你能计算出紧贴着下边沿的是哪几个正方形吗？

输出

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

思路：一共154*154个格格，把每一格都表示为小正方形的边长。DFS（）；

AC代码：

#include<stdio.h>
#include<string.h>
#include<map>
#include<stdlib.h>
#include<algorithm>
using namespace std;
#define ll long long
#define N 155
int book[N][N];
int vis[N],flag;
int a[23]= {2,5,9,11,16,17,19,21,22,24,26,30,31,33,35,36,41,50,52,61,46,47};
void h(int x,int y,int t,int n)
{
    for(int i=x; i<x+t; i++)
        for(int j=y; j<y+t; j++)
            book[i][j]=n;
}

int solve()
{
    for(int i=1; i<=154; i++)
        for(int j=1; j<=154; j++)
            if(!book[i][j])
                return 0;
    return 1;
}

void get(int &x,int &y)
{
    for(int i=1; i<=154; i++)
        for(int j=1; j<=154; j++)
            if(!book[i][j])
            {x=i,y=j;return ;}
}

int pan(int x,int y,int t)
{
    if(x+t>155||y+t>155)return 0;
    for(int i=x; i<x+t; i++)
        for(int j=y; j<y+t; j++)
            if(book[i][j])return 0;
    return 1;
}

int dfs()
{
    int x,y;
    if(solve())return 1;
    get(x,y);
    for(int i=0; i<19; i++)
    {
        if(pan(x,y,a[i]))
        {
            if(vis[i]==0)
            {
                vis[i]=1;
                h(x,y,a[i],a[i]);
                if(dfs())return 1;
                vis[i]=0;
                h(x,y,a[i],0);
            }

        }
        else break;
    }
    return 0;

}
int main()
{
    flag=0;
    memset(book,0,sizeof(book));
    memset(vis,0,sizeof(vis));
    h(1,1,47,47);
    h(1,48,46,46);
    h(1,94,61,61);
    vis[21]=1;
    vis[20]=1;
    vis[19]=1;
    dfs();
    printf("%d\n",book[154][1]);
    for(int i=2; i<=154; i++)
    {
        if(book[154][i]!=book[154][i-1])
            printf("%d\n",book[154][i]);
    }
}