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

---

数据：

思路：

看到数据，普通dfs肯定TLE，所以我们改用记忆化搜索，然后判断每块能否分开，再枚举分的每种切法，然后求出美味值，最后判断输出。

---

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int f[25][25][25][25],a[888][888],ans,x,n,m,k,t;
bool jyh[25][25][25][25],jyh2[25][25][25][25];
bool jyhdfs(int x1,int y1,int x2,int y2)//记忆化搜索
{
	if(f[x1][y1][x2][y2]==ans)return 1;//可以
	if(f[x1][y1][x2][y2]<ans)return 0;//不可以
	if(jyh[x1][y1][x2][y2])return jyh2[x1][y1][x2][y2];
	jyh[x1][y1][x2][y2]=1;//记忆化
	for(int i=x1;i<x2;i++)//枚举行
	if(jyhdfs(x1,y1,i,y2)==true&&jyhdfs(i+1,y1,x2,y2)==true)//切
	return jyh2[x1][y1][x2][y2]=1;//可以
	for(int i=y1;i<y2;i++)//枚举列
	if(jyhdfs(x1,y1,x2,i)==true&&jyhdfs(x1,i+1,x2,y2)==true)//切
	return jyh2[x1][y1][x2][y2]=1;//可以
	return jyh2[x1][y1][x2][y2]=0;//不可以
}
int main()
{
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		memset(a,0,sizeof(a));//初始化
		memset(jyh,0,sizeof(jyh));
		scanf("%d%d%d",&n,&m,&k);
		for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&x);
	  	    	a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+x;//求前缀和
		} 
		for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		for(int k=i;k<=n;k++)
		for(int l=j;l<=m;l++)
	       f[i][j][k][l]=a[k][l]-a[i-1][l]-a[k][j-1]+a[i-1][j-1];//求出每一块巧克力
		ans=f[1][1][n][m]/k;//求美味值
		if(f[1][1][n][m]%k)printf("NO\n");//判断能不能
	else if (jyhdfs(1,1,n,m)) printf("YES\n");//dfs
		else printf("NO\n");//否则输出no
	}
	
	return 0;
}