Chocolate

Chocolate

题目

Charlie 有一块巧克力。
这块巧克力是矩形的，有 n 行 m 列一共 n × m 个大小相同的小块，每一小块都有一个美味值 ai,j。
Charlie 有 k 个朋友，他希望把巧克力分给这些朋友。
Charlie 按如下方法分配巧克力：做 k-1 次分割，每次拿出一块巧克力，将它
沿水平或竖直方向分成两块矩形的巧克力。分割完成后一共有 k 块巧克力，Charlie
会把这 k 块巧克力一一分给他的朋友们。
一块巧克力的美味值定义为它的所有小块的美味值之和。Charlie 想知道是否
存在一种可行的方案，使每个朋友获得的巧克力的美味值相等。

输入

本题有多组测试数据。第一行一个正整数 T 表示数据组数。
对于每组测试数据：
第一行 3 个正整数表示 n, m, k。
接下来 n 行，每行 m 个正整数，表示每一小块的美味值

输出

对于每个测试数据，输出一行 YES 或 NO，表示是否存在可行方案。

输入样例

2
1 3 2
2 1 1
2 2 3
2 3
3 1

输出样例

YES
NO

数据范围

思路

这道题就是一道搜索题，我这里用 $dfs$ 配记忆化搜索来做。
就按着题目，有横着分和竖着分的情况，而每个人要分到的值就是整个巧克力的美味值和除以人数。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,m,k,t,z,f[25][25][25][25],a[25][25];
bool work(int x1,int y1,int x2,int y2)//dfs
{
	int tot=a[x2][y2]-a[x1-1][y2]-a[x2][y1-1]+a[x1-1][y1-1];//求出这一小块区域的美味值和
	if (f[x1][y1][x2][y2]>0) return f[x1][y1][x2][y2]%2;//记忆化搜索
	if (tot==z) {f[x1][y1][x2][y2]=1;return true;}//已经分到了要求的美味值
	if (tot%z!=0) {f[x1][y1][x2][y2]=2;return false;}//美味值不为整数（分不了）
	for (int i=x1;i<x2;i++)//横着分
	{
		if (work(i+1,y1,x2,y2)&&work(x1,y1,i,y2))//判断可不可以切
		{
			f[x1][y1][x2][y2]=1;//记忆化搜索
		    return 1;
	 	}
	}
	for (int i=y1;i<y2;i++)//竖着分
	{
		if (work(x1,i+1,x2,y2)&&work(x1,y1,x2,i))//判断可不可以切
		 {
		 	f[x1][y1][x2][y2]=1;//记忆化搜索
		    return 1;
		 } 
	}
	f[x1][y1][x2][y2]=2;//记忆化搜索
	return 0;
}
int main()
{
	scanf("%d",&t);//读入
	for (int i=1;i<=t;i++)
	{
		memset(a,0,sizeof(a));//初始化
		memset(f,0,sizeof(f));//初始化
		scanf("%d%d%d",&n,&m,&k);//读入
		for (int j=1;j<=n;j++)
		for (int k=1;k<=m;k++)
		scanf("%d",&a[j][k]);//读入
		for (int i=1;i<=n;i++)
		for (int j=1;j<=m;j++)
		a[i][j]+=a[i-1][j]+a[i][j-1]-a[i-1][j-1];//求前缀和
		z=a[n][m]/k;//求出每一个人应得到的美味值
		if (a[n][m]%k!=0) {printf("NO\n");continue;}//美味值不为整数（分不了）
		if (work(1,1,n,m)) printf("YES\n");//dfs
		else printf("NO\n");  
	}
}