【蓝桥杯】剪格子-DFS

资源限制

时间限制：1.0s   内存限制：256.0MB

问题描述

如下图所示，3 x 3 的格子中填写了一些整数。

+--*--+--+
|10* 1|52|
+--****--+
|20|30* 1|
*******--+
| 1| 2| 3|
+--+--+--+

我们沿着图中的星号线剪开，得到两个部分，每个部分的数字和都是60。

本题的要求就是请你编程判定：对给定的m x n 的格子中的整数，是否可以分割为两个部分，使得这两个区域的数字和相等。

如果存在多种解答，请输出包含左上角格子的那个区域包含的格子的最小数目。

如果无法分割，则输出 0。

输入格式

程序先读入两个整数 m n 用空格分割 (m,n<10)。

表示表格的宽度和高度。

接下来是n行，每行m个正整数，用空格分开。每个整数不大于10000。

输出格式

输出一个整数，表示在所有解中，包含左上角的分割区可能包含的最小的格子数目。

样例输入1

3 3
10 1 52
20 30 1
1 2 3

样例输出1

3

样例输入2

4 3
1 1 1 1
1 30 80 2
1 1 1 100

样例输出2

10

 

#include <iostream>
#include <string.h>
using namespace std;
int a[10][10];
int dx[4] = {0,1,0,-1};
int dy[4] = {1,0,-1,0};
bool isv[10][10];
int M,N,SUM;
bool judge(int x,int y,int num)
{
    if( x<1 || y<1 || x>N || y>M )    //越界 
        return 1;
    if( isv[x][y] )    //访问过 
        return 1;
    if( num + a[x][y] > SUM/2 )    //走这一步超过了和的1半 
        return 1;
    return 0;
}
int dfs(int x,int y,int num)
{
    if(num == SUM/2){
        return 1;
    }
    for(int i=0;i<4;i++){
        int nx = x + dx[i];
        int ny = y + dy[i];
        if(judge(nx,ny,num))    //判断 
            continue;
        //下一步可以走
        isv[nx][ny] = true;
        int res = dfs(nx,ny,num+a[nx][ny]);
        if(res)    //产生结果，直接返回 
            return res+1;
        isv[nx][ny] = false;
    }
    return 0; 
}
int main()
{
    while(cin >> M >> N){
        SUM = 0;
        for(int i = 1;i <= N;i++)
            for(int j = 1;j <= M;j++){
                cin >> a[i][j];
                SUM += a[i][j];
            }
        if( SUM%2 ){    //和是奇数一定不可以 
            cout << 0 << endl;
        }
        else{    //和是偶数继续判断
            memset(isv,0,sizeof(isv));
            isv[1][1] = true;
            cout<<dfs(1,1,a[1][1])<<endl;
        }
    }
    return 0;
}