代码随想录补充题:44. 开发商购买土地

67 阅读2分钟
  1. 开发商购买土地(第五期模拟笔试)
题目描述

在一个城市区域内,被划分成了n * m个连续的区块,每个区块都拥有不同的权值,代表着其土地价值。目前,有两家开发公司,A 公司和 B 公司,希望购买这个城市区域的土地。 

现在,需要将这个城市区域的所有区块分配给 A 公司和 B 公司。

然而,由于城市规划的限制,只允许将区域按横向或纵向划分成两个子区域,而且每个子区域都必须包含一个或多个区块。 为了确保公平竞争,你需要找到一种分配方式,使得 A 公司和 B 公司各自的子区域内的土地总价值之差最小。 

注意:区块不可再分。

输入描述

第一行输入两个正整数,代表 n 和 m。 

接下来的 n 行,每行输出 m 个正整数。

输出描述

请输出一个整数,代表两个子区域内土地总价值之间的最小差距。

输入示例
3 3
1 2 3
2 1 3
1 2 3
输出示例
0
提示信息

如果将区域按照如下方式划分:

1 2 | 3
2 1 | 3
1 2 | 3 

两个子区域内土地总价值之间的最小差距可以达到 0。

数据范围:

1 <= n, m <= 100;
n 和 m 不同时为 1。

题解: 用前缀和方法,处理好代码细节,问题不大,这里数组用了增大一个位的技巧,正好用来存前缀和。 前缀和计算的时候可以用迭代的方法,降低计算复杂度。

#include <stdio.h>
#include <stdlib.h>

int main()
{
    int n, m, i, j, min;
    scanf("%d %d", &n, &m);
    int** land = (int**)malloc(sizeof(int*) * n + 1);
    for (i = 0;i <= n;i ++)
    {
        land[i] = (int*)malloc(sizeof(int) * m + 1);
    }
    land[0][0] = 0;
    for (i = 1;i <= n;i ++)
    {
        land[i][0] = 0;
        for(j = 1;j <= m;j ++)
        {
            scanf("%d", &land[i][j]);
            land[i][0] = land[i][0] + land[i][j];
        }
        land[i][0] = land[i][0] + land[i-1][0];
    }

    for (j = 1;j <= m;j ++)
    {
        land[0][j] = 0;
        for (i = 1;i <= n;i ++)
        {
            land[0][j] = land[0][j] + land[i][j];
        }
        land[0][j] = land[0][j] + land[0][j - 1];
    }
    min = land[n][0];
    for (i = 0;i <= n;i ++)
    {
        if (min > abs(land[n][0] - 2 * land[i][0]))
        min = abs(land[n][0] - 2 * land[i][0]);
    }
    for (j = 0;j <= m;j ++)
    {
        if (min > abs(land[0][m] - 2 * land[0][j]))
        min = abs(land[0][m] - 2 * land[0][j]);
    }
    printf("%d", min);



    return 0;
}