题意
题目大意是求一个大的正方形蛋糕能否切分成n(1<=n<=16)个指定边长t(1<=t<=10)的小正方形蛋糕。
分析
由边长可推出原大蛋糕的边长最大为40。用二维数组来递归必然会超时,简化一下,用一维数组来表示蛋糕。
想像一下,用一个一维数组来描述一个正方形,数组的值表示每个位置已经被填充的高度。每次填充时,找到最左边的最低位置。
比如说第二个输入样本,大蛋糕的边长为5,小蛋糕最大的边长为3,那么首先找到坐标0的位置,这里的高度为0,而且1,2位置的高度也为0,那么表示这个小蛋糕可以填在这里,放好后,0,1,2位置的高度均为3。
第二块小蛋糕的边长也为3,找到最左边的最低位置3(此时还是0)。此时,从3到右边界4,只有两个位置,因此不能放这里。
接着判断第3块蛋糕,以此类推。。。最终可找到是否能完整填充。
代码如下:
Memory: 236K Time: 16MS Length:69LINES
#include<iostream>
using namespace std;
int sides[11] = {}; //不同边长(1-10)的小蛋糕数量
int filled[40] = {}; //简化成一维数组后,每个位置的高度
int side = 0; //大蛋糕的边长
bool search(int count)
{
if (count == 0) return true;
int st = side;
int pos = 0;
for (int j = 0; j < side; ++j) //寻找高度最低的位置
{
if (filled[j] < st)
{
st = filled[j];
pos = j;
}
}
for (int i = 10; i > 0; --i) //从大到小填充
{
if (sides[i] > 0 && pos + i - 1 < side && i + st - 1 < side)
{
int j = pos;
for (; j < pos + i; ++j) if (filled[j] > st) break;//判断后面i(边长)个位置是否一样高
if (j == pos + i)
{
for (int k = pos; k < pos + i; ++k) filled[k] += i;//填充
--sides[i];
if (!search(count - 1)) //递归&&回溯
{
for (int k = pos; k < pos + i; ++k) filled[k] -= i;
++sides[i];
}
else return true;
}
}
}
return false;
}
int main()
{
int cases = 0;
int count = 0;
cin >> cases;
for (int i = 0; i < cases; ++i)
{
cin >> side >> count;
memset(filled, 0, 40 * sizeof(int));
memset(sides, 0, 11 * sizeof(int));
int area = 0;
int tmp = 0;
for (int j = 0; j < count; ++j)
{
cin >> tmp;
++sides[tmp];
area += tmp * tmp;
}
if (area == side * side)
{
if (search(count))
{
cout << "KHOOOOB!" << endl;
continue;
}
}
cout << "HUTUTU!" << endl;
}
return 0;
}