[原创]我的PKU ACM POJ 1020解答

1
10 14 1 1 1 1 1 4 4 3 3 3 3 3 3 3

答案是KHOOOOB!

用贪心法是无法解决上面的数据的。

因此，网上参照了DFS搜索弄出了一个AC的代码，基本思路就是从左上开始切，寻找某列已切行数最少的开始，然后从大到小逐个

试蛋糕，看是否能切出来，若当前切割失败，则回溯上一步，换个不同大小的继续试。切完需要的块数就表示刚好可以。

#include <stdio.h>
#include <memory.h>
int cakeside,piecenum,pieces[11];
//fillednum stand for the nums had filled
bool fill(int fillednum,int* used)
{
//如果已填的个数等于需求的蛋糕个数则表示刚好
if(fillednum==piecenum)
return true;
int i,j,minusedrow=41,nowcol=0;
//寻找未填的行数最小的那列
for(i=1;i<=cakeside;i++)
{
if(used[i]<minusedrow)
{
minusedrow = used[i];
nowcol = i;
}
}
//从行数最小的那列开始，从大到小搜索能填入的蛋糕
for(i=10;i>0;--i)
{
if(pieces[i]>0 && i+minusedrow<=cakeside+1 && i+nowcol<=cakeside+1)
{
bool flag=true;
//判断连续的区域是否能放下当前蛋糕
for(j=nowcol;j<nowcol+i;j++)
{
if(used[j]>minusedrow)
{
flag=false;
break;
}
}
if(flag)
{
//能放下则更新蛋糕所在的列的已填行数
for(j=nowcol;j<nowcol+i;j++)
used[j] += i;
//当前大小蛋糕数减一
pieces[i]--;
//继续切下一块蛋糕
if(fill(fillednum+1,used))
return true;
//到这说明上面方案不可行，回溯，不能切当前大小的蛋糕
pieces[i]++;
for(j=nowcol;j<nowcol+i;j++)
used[j] -= i;
}
}
}
return false;
}
int main()
{
int ncase,i,j;
scanf("%d",&ncase);
while(ncase>0)
{
--ncase;
int sum=0;
int pieceside;
scanf("%d %d",&cakeside,&piecenum);
memset(pieces,0,sizeof(pieces));
//记录各不同大小的蛋糕个数，大小范围在1-10之间
for(i=0;i<piecenum;++i)
{
scanf("%d",&pieceside);
//大小为pieceside的个数
pieces[pieceside]++;
sum += pieceside*pieceside;
}
if(sum!=cakeside*cakeside)
{
printf("HUTUTU!\n");
continue;
}
bool IsWaste=true;
//used[i]记录的是第i列还没填的那行
int used[41];
for(i=1;i<=40;i++) used[i]=1;
//fill表示从第几块蛋糕开始填是否能刚好填满所有蛋糕
IsWaste = !fill(0,used);
if(IsWaste)
printf("HUTUTU!\n");
else
printf("KHOOOOB!\n");

}
return 0;
}