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 对数据进行处理。