POJ 1020--Packets

题意

题目大意是求一个大的正方形蛋糕能否切分成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;
}