ABC159 E. Dividing Chocolate

Question

一块由‘0’、‘1’组成的$H\times W$的巧克力，将其进行横着切或者竖着切(行与行之间、列与列之间)，使得切完后的每一块不超过$K$块’1’。

Solution

这道题目的关键在于$H$的范围很小。我们可以枚举横着切的方式，在横着切的基础之上对竖着切进行贪心。
__builtin_popcount()用于计算一个 32 位无符号整数有多少个位为1

Code

#include<bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const double eps = 1e-8;
const int NINF = 0xc0c0c0c0;
const int INF  = 0x3f3f3f3f;
const ll  mod  = 1e9 + 7;
const ll  N = 1e3 + 5;

int n,m,k,z;
string s[15];
int c[15];

bool f(int j){//返回true代表要切一刀竖着的，因为至少有一块已经超过k了
	int id=0;
	for(int i=0;i<n;i++){
		c[id]+=s[i][j]=='1';
		if(c[id]>k) return true;//超过k了 要切哦
		if(z>>i & 1) id++;
	}
	return false;//嘿嘿 不用切 依旧满足
}

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>m>>k;
	for(int i=0;i<n;i++) cin>>s[i];
	int ans=0x3f3f3f3f;
	for(z=0;z<1<<n-1;z++){//长度为n那么可以切n-1刀，枚举0~2^(n-1)-1的状态，1代表切，0代表不切。
		memset(c,0,sizeof(c));
		bool ok=false;
		int cnt=__builtin_popcount(z);//横着切了多少刀
		for(int j=0;j<m;j++){//现在横着切完要贪心求竖着了
			if(f(j)){//要竖着切一刀
				cnt++;//竖着切了一刀
				memset(c,0,sizeof(c));//那么切完之后竖着的每一块又都是0了
				ok=f(j);//如果竖着切了一刀，哪怕是一列还是不行，那说明横着切少了，无论如何都不满足
			}
		}
		if(!ok) ans=min(ans,cnt);//满足条件的时候记录最小值
	}
	cout<<ans;
	return 0;
}