AtCoder Beginner Contest 159 E - Dividing Chocolate 题解（思维+枚举+状态压缩）

题目链接

题目思路

其实要观察这个题目的数据范围，会发现有蹊跷

行数那么少,暴力搜行的切割方法.然后扫描列,对于一定要切的地方就切.复杂度O(2^n *(n∗m))

注意枚举行的状态只有1<<(h-1) 因为最多切割h-1条

代码

#include<cstdio>
#include<algorithm>
using namespace std;
const int inf=0x3f3f3f3f;
int h,w,k,a[20][1010],pre[20][1010],flag,ans=inf;
char ch;
int main(){
    scanf("%d %d %d",&h,&w,&k);
    for(int i=1;i<=h;i++){
        for(int j=1;j<=w;j++){
            scanf(" %c",&ch);
            pre[i][j]=pre[i][j-1]+ch-'0';
        }
    }
    for(int state=0;state<(1<<h-1);state++){
        int last=0,sum=0,cnt=0,flag=1;
        for(int j=1;j<=w;j++){
            for(int i=1;i<=h;i++){
                sum=sum+pre[i][j]-pre[i][last];
                if(sum>k){
                    if(j-last==1){//只有一列不能再切不满足条件
                        flag=0;
                        break;
                    }
                    i=0,sum=0,last=j-1;//第二轮循环再来一次
                    cnt++;
                }
                if(i==h||state&(1<<(i-1))){
                    sum=0;
                }
            }
            if(!flag){
                break;
            }
        }
        if(flag){
            ans=min(ans, __builtin_popcountl(state)+cnt);
        }
    }
    printf("%d\n",ans);
    return 0;
}