(poj 1020 Anniversary Cake)<DFS的巧妙剪枝>

来源:互联网 发布:雷蛇驱动有mac版吗 编辑:程序博客网 时间:2024/05/16 14:30

传送门


solution

一开始看到数据范围很小,直接二维vis数组暴力模拟是否覆盖
出乎意料的TLE qwq
然后看题解,才发现自己还是naive

  • 一维数组cov[]表示每一列从下到上已经覆盖的格子数
  • 将小正方形从小到大排序,因为小的更灵活,因此先确定大的位置
  • 每次都卡着左下角摆,宁左勿右 #滑稽
  • 重要剪枝:如果某个正方形被摆在第一列(即最左的最小占用列就是第一列),但剩下的正方形没有合法方案,那么这种方案就可以淘汰了。因为此时的正方形是剩下的正方形中最大的,把它先放在当前位置都无法满足条件,那么其它更小更灵活的正方形被放好后它肯定还是找不到合适位置,直接break
  • 还有一个很多题解没有提到的剪枝:
if(s-minl<a[i]) break;

即找到的最大可占用的空间都无法满足当前正方形,那么放好其它比它小的正方形后依然无法为它找到合适位置,说明这种方案不可行,可以break
但是if(x+a[i]-1>s) continue; 此处只能continue,原因:虽然此时找的x是最小覆盖的列,但是这个正方形未必要放在“最宽敞”的地方,因此可以先确定比它小的其它正方形,然后可能还会留下合适的位置


Code

// by spli#include<cstring>#include<algorithm>#include<iostream>#include<cstdio>using namespace std;int T;int s,n;int a[20];bool vis[20];int cov[20];bool cmp(int x,int y){    return x>y;}bool dfs(int k){    if(k==n+1) return 1;    int minl=0x3f3f3f3f,x=0;    for(int i=1;i<=s;++i)        if(minl>cov[i]){            minl=cov[i];            x=i;        }    for(int i=1;i<=n;++i){        if(vis[i]) continue;        if(s-minl<a[i]) break;        if(x+a[i]-1>s) continue;         int j=x;        for(;j<=s;++j)            if(cov[j]!=minl) break;        j--;        if(j-x+1>=a[i]){            vis[i]=1;            for(int c=x;c<=x+a[i]-1;++c) cov[c]+=a[i];              if(dfs(k+1)) return 1;                  for(int c=x;c<=x+a[i]-1;++c) cov[c]-=a[i];              vis[i]=0;               }        while(a[i+1]==a[i]) i++;    }    return 0;}int main(){    scanf("%d",&T);    while(T--){        int sum=0;        scanf("%d%d",&s,&n);        for(int i=1;i<=n;++i) scanf("%d",&a[i]),sum+=a[i]*a[i];        if(sum!=s*s){            puts("HUTUTU!");            continue;        }        sort(a+1,a+1+n,cmp);        for(int i=1;i<=s;++i) cov[i]=0;        for(int i=1;i<=n;++i) vis[i]=0;        if(dfs(1)) puts("KHOOOOB!");        else puts("HUTUTU!");    }    return 0;}
原创粉丝点击