前缀和解决问题:区间和
基本思想:使用一个数组a[n],a[i]表示前i个元素的和,那么要计算i+1~n号元素和时就可以使用如下方法:a[n-1] - a[i]
58. 区间和
题目链接:58. 区间和(第九期模拟笔试)
题目描述:
给定一个整数数组 Array,请计算该数组在每个指定区间内元素的总和。
解题思路:
前缀和经典题目
代码:
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int n;
cin >> n;
vector<int> arr(n+1);
for(int i = 1; i <= n; i++)
{
int x = 0;
cin >> x;
arr[i] = arr[i-1] + x;
}
int a, b;
while(scanf("%d%d",&a,&b) != EOF)
{
cout << arr[b+1] - arr[a] <<endl;
}
return 0;
}
44. 开发商购买土地
题目链接:44. 开发商购买土地(第五期模拟笔试)
题目描述:
在一个城市区域内,被划分成了n * m个连续的区块,每个区块都拥有不同的权值,代表着其土地价值。目前,有两家开发公司,A 公司和 B 公司,希望购买这个城市区域的土地。
现在,需要将这个城市区域的所有区块分配给 A 公司和 B 公司。
然而,由于城市规划的限制,只允许将区域按横向或纵向划分成两个子区域,而且每个子区域都必须包含一个或多个区块。 为了确保公平竞争,你需要找到一种分配方式,使得 A 公司和 B 公司各自的子区域内的土地总价值之差最小。
注意:区块不可再分。
解题思路:
前缀和求解:横向使用一个前缀和数组,纵向使用一个前缀和数组,之后就是按照前缀和思路求解最小差值
代码:
#include<iostream>
#include<vector>
#include <climits>
using namespace std;
int main()
{
int n, m;
cin >> n >> m;
vector<vector<int>> matrix(n,vector<int>(m));
vector<int> col(m+1);
vector<int> row(n+1);
for(int i = 0; i < n ; i++)
{
int sum = 0;
for(int j = 0; j < m; j++)
{
cin >> matrix[i][j];
sum += matrix[i][j];
}
row[i+1] =row[i] + sum;
}
for(int j = 0; j < m; j++)
{
int sum = 0;
for(int i = 0; i < n; i++)
{
sum += matrix[i][j];
}
col[j+1] = col[j] + sum;
}
int result = INT_MAX;
for(int i = 1; i < m; i++)
result = min(result, abs(col[m] - 2 * col[i]));
for(int i = 1; i < n; i++)
result = min(result, abs(row[n] - 2 * row[i]));
cout << result << endl;
return 0;
}