代码随想录 -- 数组 -- 开发商购买土地

44. 开发商购买土地（第五期模拟笔试） (kamacoder.com)

借助前缀和的思想。

思路：定义两个数组，一个存每一行的和，一个存每一列的和。最后遍历这两个数组时借助前缀和减少计算量。

import sys

input=sys.stdin.read
data=list(map(int, input().split()))

index=0
n=data[index]
index+=1
m=data[index]
index+=1

arr=[[0]*m for i in range(n)]
sumRow=[0]*n
sumC=[0]*m
Sum=0
for i in range(n):
    sum=0
    for j in range(m):
        arr[i][j]=data[index]
        sum=sum+arr[i][j]
        Sum=Sum+arr[i][j]
        index+=1
    sumRow[i]=sum

for j in range(m):
    sum = 0
    for i in range(n):
        sum += arr[i][j]
    sumC[j] = sum

minSub=float('inf')
row=0
for i in range(n):
    row=row+sumRow[i]
    minSub=min(abs(Sum-2*row), minSub)
c=0
for i in range(m):
    c=c+sumC[i]
    minSub=min(abs(Sum-2*c), minSub)
    
if n==1 or m==1:
    minSub=abs(data[2]-data[3])
print(minSub)

注意当行或列为1时要特殊处理！ 

知识点

ACM输入输出模式

#读取所有输入数据
input = sys.stdin.read()
input()

当命令行中输入多行文本时，可以使用 input = sys.stdin.read() 来读取所有数据，并使用 input().split() 进行分割。

sys.stdin.read() 是 sys 模块中的一个函数，用于读取标准输入（通常是键盘输入）的所有内容，并将其作为一个字符串返回。

后续可以借助 map() 函数和 index 对数据进行处理。