【代码随想录】刷题记录(5)-开发商购买土地

【题目描述】

在一个城市区域内，被划分成了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

 

我的作答：

虽然看着很复杂，实际上理解题意之后，就是按矩阵的行和列切开，没有过多的技巧，类似于区间和的变体，把一维数组改成了二维数组来存储元素而已。

import sys

def gapMin():
    inpuut = sys.stdin.read()
    data = inpuut.split()
    n = int(data[0])
    m = int(data[1])
    count = 2  #索引向量，用于之后从data中读数据到二维数组里
    matrix = [[0]*m for a in range(0, n)]
    matrixSum = [[0]*m for b in range(0, n)]
    iGap = [0*n for c in range(n)]
    jGap = [0*m for d in range(m)]
    for i in range(0, n):
        for j in range(0, m):
            matrix[i][j] = int(data[count])  #[0]和[1]是n和m，所以从[2]开始存数据
            if j==0&i==0:
                matrixSum[i][j] = matrix[i][j] #递归一定要分=0的情况！！
            elif j==0:
                matrixSum[i][j] = matrixSum[i-1][j-1]+matrix[i][j]
            else:
                matrix[i][j] = matrixSum[i][j-1]+matrix[i][j]
            count += 1
    
    for k in range(n):
        iGap[k] = abs(2*matrixSum[k][-1]-matrixSum[-1][-1]) #寻找按行切割的最小gap
    imin = min(iGap)
    
    for g in range(m): #寻找按列切割的最小gap，先用一维数组存储所有列gap值
        jGap[g] = abs(2*sum(matrixSum[row][g] for row in range(n))-matrixSum[-1][-1])
    jmin = min(jGap)
    
    return min(imin, jmin)
    
if __name__=='__main__':
    gapmin = gapMin()
    print(gapmin)

出的问题还是递归的时候没有分清楚条件，特别是现在是两重循环，要分别区分每重循环初始的时候的情况！！非常容易出错！！

参考代码：

def main():
    import sys
    input = sys.stdin.read
    data = input().split()

    idx = 0
    n = int(data[idx])
    idx += 1
    m = int(data[idx])
    idx += 1
    sum = 0
    vec = []
    for i in range(n):
        row = []
        for j in range(m):
            num = int(data[idx])
            idx += 1
            row.append(num)
            sum += num
        vec.append(row)

    # 统计横向
    horizontal = [0] * n
    for i in range(n):
        for j in range(m):
            horizontal[i] += vec[i][j]

    # 统计纵向
    vertical = [0] * m
    for j in range(m):
        for i in range(n):
            vertical[j] += vec[i][j]

    result = float('inf')
    horizontalCut = 0
    for i in range(n):
        horizontalCut += horizontal[i]
        result = min(result, abs(sum - 2 * horizontalCut))

    verticalCut = 0
    for j in range(m):
        verticalCut += vertical[j]
        result = min(result, abs(sum - 2 * verticalCut))

    print(result)

if __name__ == "__main__":
    main()

可以下次试试append函数，看着很简洁；而且如果设置idx索引变量，每次++的话，就不用自己手动计算需要赋值的时候idx究竟是多少了！