SSLOJ 1314 Chocolate
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;
}