【蓝桥杯】第六届国赛C语言B组 2.完美正方形（dfs）

 1 #include <cstdio>
 2 #include <cmath>
 3 #include <iostream>
 4 #include <cstring>
 5 #include <algorithm>
 6 using namespace std;
 7 int N,vis[60];
 8 int d[] = {2,5,9,11,16,17,19,21,22,24,26,30,31,33,35,36,41,50,52,46,47,61};
 9 int G[155][155];
10 
11 // 判断是否可以放置边长为n的正方形
12 bool inside(int x,int y,int n)
13 {
14     if(!(x+n <= 155 && y+n <= 155))
15         return false;
16     for(int i = x;i < n+x;i++)
17         for(int j = y;j < n+y;j++)
18             if(G[i][j])
19                 return false;
20     return true;
21 }
22 
23 // 将边长为n的正方形放进去或者拿走该正方形
24 // m==0 取走   m==n 放置
25 void Fill(int x,int y,int n,int m)
26 {
27     for(int i = x;i < n+x;i++)
28         for(int j = y;j < n+y;j++)
29             G[i][j] = m;
30 }
31 
32 // 找到可以放置的坐标
33 void Get(int &x,int &y)
34 {
35     for(int i = 1;i <= 154;i++)
36         for(int j = 1;j <= 154;j++)
37             if(!G[i][j])
38             {
39                 x = i;
40                 y = j;
41                 return ;
42             }
43 }
44 
45 // 检查大正方形是否放满
46 bool solved()
47 {
48     for(int i = 1;i <= 154;i++)
49         for(int j = 1;j <= 154;j++)
50             if(!G[i][j])
51                 return false;
52     return true;
53 }
54 
55 bool dfs(int x,int y)
56 {
57     if(solved())
58         return true;
59     else
60     {
61         Get(x,y);
62         for(int k = 0;k < 19;k++)
63         {
64             if(inside(x,y,d[k]))
65             {
66                 if(!vis[k])
67                 {
68                     Fill(x,y,d[k],d[k]);
69                     vis[k] = 1;
70                     if(dfs(x,y+d[k]))
71                         return true;
72                     Fill(x,y,d[k],0);
73                     vis[k] = 0;
74                 }
75             }
76             else
77                 break;
78         }
79     }
80     return false;
81 }
82 
83 int main()
84 {    
85     // int a[30];
86     memset(vis,0,sizeof(vis));
87     memset(G,0,sizeof(G));
88     Fill(1,1,47,47);
89     Fill(1,48,46,46);
90     Fill(1,94,61,61);
91     sort(d,d+19);
92     dfs(1,1);
93 
94     for(int i = 1;i <= 154;i++)
95         cout << G[154][i] << ‘ ‘;
96 
97     return 0;
98 }