SSLOJ·Chocolate【DFS】

Description–

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

---

Input–

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

Output–

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

---

Sample Input–

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

Sample Output–

YES
NO

---

说明–

---

代码–

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
int t,n,m,k,u,s;
int a[30][30],b[30][30];
bool pd[30][30][30][30],f[30][30][30][30];
bool dfs(int x,int y,int xr,int yr)
{
	if (pd[x][y][xr][yr]) 
	  return f[x][y][xr][yr];
	s=b[xr][yr]-b[x-1][yr]-b[xr][y-1]+b[x-1][y-1];
	pd[x][y][xr][yr]=1;
	if (s%u!=0)
	  return 0;
	if (s==u)
	  return f[x][y][xr][yr]=1;
	for (int i=x;i<xr;++i)
	  if (dfs(x,y,i,yr) && dfs(i+1,y,xr,yr))
	  	return f[x][y][xr][yr]=1;
	for (int i=y;i<yr;++i)
	  if (dfs(x,y,xr,i) && dfs(x,i+1,xr,yr))
	    return f[x][y][xr][yr]=1;
}
int main()
{
	scanf("%d",&t);
	for (int i=1;i<=t;++i)
	{
	    memset(f,0,sizeof(f));
	    memset(pd,0,sizeof(pd));
	    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]);
	      	  b[i][j]=b[i-1][j]+b[i][j-1]-b[i-1][j-1]+a[i][j];//前缀和
	      }
	    if (b[n][m]%k!=0)
		{
			printf("NO\n");
			continue;
		}
	    u=b[n][m]/k;//目标美味值
	    if (!dfs(1,1,n,m))
	      printf("NO\n");
	    else
	      printf("YES\n");
	}
	
	return 0;
}