【代码随想录】开发商购买土地——前缀和思想

本博文为《代码随想录》学习笔记，原文链接为：44. 开发商购买土地 | 代码随想录

题目

原题链接：44. 开发商购买土地（第五期模拟笔试）

【题目描述】

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

思路

只允许按照横向或纵向划分子区域，因此先分别统计每一行和每一列的权值。

在计算区域上半部分（或左半部分）权值和时，采用前缀和思想，使用horizontalCut（或veticalCut）逐行相加（或逐列相加）。sum-horizontalCut（或sum-verticalCut）即为另一半区域的权值和，两部分相减的绝对值取最小。

题解

#include <bits/stdc++.h>
using namespace std;

int main()
{
	int n,m;
	int result=INT32_MAX,sum=0;
	cin>>n>>m;
	vector<vector<int>> map(n,vector<int>(m,0));
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			cin>>map[i][j];
			sum+=map[i][j];
		}
	}
	vector<int>horizontal(n,0);//统计每一行的权值 
	for(int i=0;i<n;i++)
	{
		for(int j=0;j<m;j++)
		{
			horizontal[i]+=map[i][j]; 
		}
	}
	vector<int>vertical(m,0);//统计每一列的权值 
	for(int j=0;j<m;j++)
	{
		for(int i=0;i<n;i++)
		{
			vertical[j]+=map[i][j]; 
		}
	}
	int horizontalCut=0;
	for(int i=0;i<n;i++)
	{
		
	    horizontalCut+=horizontal[i];//行前缀和 ，即上半部分权值 
	    result=min(result,abs(sum-horizontalCut-horizontalCut));//sum-horizontalCut为下半部分权值，horizontalCut为上半部分权值 
	}
	int verticalCut=0;
	for(int i=0;i<n;i++)
	{
		
	    verticalCut+=vertical[i];//列前缀和 ，即左半部分权值 
	    result=min(result,abs(sum-verticalCut-verticalCut));//sum-verticalCut为下半部分权值，verticalCut为上半部分权值 
	}
	cout<<result<<endl;
	return 0; 
}

结果