题目大意:有一个大小为s*s的大正方形,给你n个小正方形,第i个小正方形的大小为ai*ai,问是否存在一种方案,使得这些小正方形正好填充大正方形。
思路:这是一道深搜+回溯+贪心题。在这道题中,贪心策略为每次选择被占用最少的地方,从大到小进行尝试。由于有回溯的存在,因此贪心策略的正确性貌似是对的,不过我始终没有能够严格的证明。另外,深搜除了常规剪枝之外,还可以加一个剪枝,即第一块小正方形如果没有成功,则肯定没有方案,剪枝。
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
bool flag;
int original_size,column_occupy[42],cake[11],n;
void dfs(int remain_cake)
{
int min_column,min_pos,i,j;
if (remain_cake==0)
{
flag=true;
return ;
}
for (i=1,min_column=42; i<=original_size; i++)
if (column_occupy[i]<min_column)
{
min_column=column_occupy[i];
min_pos=i;
}
for (i=10; i>=1 && flag==false; i--)
{
if (cake[i]>0 && min_pos+i-1<=original_size && min_column+i<=original_size)
{
for (j=min_pos; j<min_pos+i; j++)
if (column_occupy[j]>min_column)
break;
if (j==min_pos+i)
{
cake[i]--;
for (j=min_pos; j<min_pos+i; j++)
column_occupy[j]+=i;
dfs(remain_cake-1);
for (j=min_pos; j<min_pos+i; j++)
column_occupy[j]-=i;
cake[i]++;
}
if (remain_cake==n)
return ;
}
}
}
int main()
{
int test_cases,i,sum_area,tmp;
scanf("%d",&test_cases);
while (test_cases--)
{
memset(cake,0,sizeof(cake));
memset(column_occupy,0,sizeof(column_occupy));
scanf("%d%d",&original_size,&n);
for (i=1,sum_area=0; i<=n; i++)
{
scanf("%d",&tmp);
cake[tmp]++;
sum_area+=tmp*tmp;
}
if (sum_area!=original_size*original_size)
{
printf("HUTUTU!\n");
continue;
}
flag=false;
dfs(n);
if (flag)
printf("KHOOOOB!\n");
else
printf("HUTUTU!\n");
}
return 0;
}