蓝桥杯练习系统历届试题 剪格子 dfs

问题描述

如下图所示，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

 

果然赤果果的dfs回溯一些就过了。

附代码：

/*
 好像是很直接的dfs+回溯
 */

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <queue>
#define inf 100000000
using namespace std;

int mp[20][20];
int ans, sum;
int n, m;

int dir[4][2] = {1, 0, -1, 0, 0, 1, 0, -1};
int vis[20][20];

bool check(int a, int b) {
    if (a >= 0 && a < n && b >= 0 && b < m && !vis[a][b]) {
        return true;
    }
    return false;
}

void dfs(int x, int y, int num, int tsum) {
    if (tsum == sum/2) {
        if (ans > num) {
            ans = num;
            return;
        }
    }

    for (int i=0; i<4; ++i) {
        int tx = x + dir[i][0];
        int ty = y + dir[i][1];
        if (check(tx, ty)) {
            vis[tx][ty] = 1;
            dfs(tx, ty, num+1, tsum+mp[tx][ty]);
            vis[tx][ty] = 0;
        }
    }
}

int main() {
    while(cin >> m >> n) {
        ans = inf;
        sum = 0;
        memset(vis, 0, sizeof(vis));

        for (int i=0; i<n; ++i) {
            for (int j=0; j<m; ++j) {
                cin >> mp[i][j];
                sum += mp[i][j];
            }
        }

        vis[0][0] = 1;
        dfs(0, 0, 1, mp[0][0]);
        cout << ans << endl;
    }
    return 0;
}