补充:前缀和篇--代码随想录算法训练营第三十四天| 58. 区间和,44. 开发商购买土地

前缀和解决问题:区间和

基本思想:使用一个数组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;
}

  • 6
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值