【记忆化搜索】【DFS】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

思路

记忆化搜索
详见代码

#include<Iostream>
#include<Cstring>
#include<Cstdio>
using namespace std;
bool Bol[30][30][30][30];
bool B[30][30][30][30];
int F[30][30][30][30];
int A[30][30];
int T,Sum;
int n,m,k;
bool Dfs(int r,int c,int x,int y)
{
	if(F[r][c][x][y]==Sum)return 1;//如果达到最后所需的值
	else if(F[r][c][x][y]<Sum)return 0;//如果比平均值小
	else if(B[r][c][x][y])return Bol[r][c][x][y];//记忆化
	B[r][c][x][y]=1;//记忆化
	for(int i=c;i<y;++i)//竖着切
		if(Dfs(r,c,x,i) && Dfs(r,i+1,x,y))
			return Bol[r][c][x][y]=1;
	for(int i=r;i<x;++i)//横着切
		if(Dfs(r,c,i,y) && Dfs(i+1,c,x,y))
			return Bol[r][c][x][y]=1;
	return Bol[r][c][x][y]=0;
} 
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		memset(Bol,0,sizeof(Bol));
		memset(B,0,sizeof(B));
		scanf("%d%d%d",&n,&m,&k);
		for(int i=1;i<=n;++i)
			for(int j=1;j<=m;++j)
			{
				scanf("%d",&A[i][j]);
				A[i][j]+=A[i][j-1]+A[i-1][j]-A[i-1][j-1];//求前缀和
			}
		for(int i=1;i<=n;++i)
			for(int j=1;j<=m;++j)
				for(int x=i;x<=n;++x)
					for(int y=j;y<=m;++y)
						F[i][j][x][y]=A[x][y]-A[i-1][y]-
						A[x][j-1]+A[i-1][j-1];//求某一块的和
		Sum=F[1][1][n][m]/k;//最终每一块的和
		if(F[1][1][n][m]%k)printf("NO\n");//如果无法平均分
		else if(Dfs(1,1,n,m))printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}