代码随想录算法训练营第3天 | 第一章 数组part02 解决前一篇的疑问，区间和、前缀和以及开发商购买土地问题

解决前一篇的疑问

前一篇文章结尾代码随想录算法训练营第2天 | 第一章 数组part02 209.长度最小的子数组 59.螺旋矩阵II-CSDN博客对leetcode59.螺旋矩阵II问题我自己写的代码提出了超时不能解的疑问，现在有思路了。

详见下面：

附上现在可以通过的"循环不变量原则"的正确代码如下：

class Solution {
public:
    vector<vector<int>> generateMatrix(int n) {
        int t = 0;      //top
        int b = n-1;    //bottle
        int l = 0;      //left
        int r = n-1;    //right
        int k = 1;
        vector<vector<int>> ans(n,vector<int>(n));
        while(k<n*n){//t<=b&&l<=r
            for(int i=l;i<r;i++,k++){ans[t][i]=k;
            cout<<k<<" ";}cout<<endl;
            for(int i=t;i<b;i++,k++){ans[i][r]=k;
            cout<<k<<" ";}cout<<endl;
            for(int i=r;i>l;i--,k++){ans[b][i]=k;
            cout<<k<<" ";}cout<<endl;
            for(int i=b;i>t;i--,k++){ans[i][l]=k;
            cout<<k<<" ";}cout<<endl;
            t++;r--;b--;l++;
            cout<<"t:"<<t<<"r:"<<r<<"b:"<<b<<"l:"<<l<<endl;
        }
        if(n%2)ans[n/2][n/2]=k;
        return ans;
    }
};

区间和、前缀和

前缀和是一种思维巧妙很实用 而且 很有容易理解的一种算法思想，大家可以体会一下

文章讲解：58. 区间和 | 代码随想录

可以通过下面图示简单理解一下：

 主要就是因为要不停给出区间长度，然后计算出区间中的元素和。最暴力的想法是两层for循环，优化解法便是用前缀和的思想，

#include<iostream>
using namespace std;
#include<vector>

int main(){
    int n;
    cin>>n;
    vector<int> vec(n);
    vector<int> p(n);
     
    //边输入边初始化前缀和
    int presum=0;
    for(int i=0;i<n;i++){
        scanf("%d",&vec[i]);
        presum+=vec[i];
        p[i]=presum;
    }
    
    int a,b;//所求区间上下界
    while(scanf("%d%d",&a,&b)!=EOF){
        int sum=0;
        if(a==0)sum=p[b];
        else sum=p[b]-p[a-1];
        printf("%d\n",sum);
    }
    
    
}

初始化前缀和，后面要求的不过是前缀和数组中指定元素的差。

上述有个值得注意的地方就是while(scanf("%d%d",&a,&b)!=EOF)这个在官方解答中给出的是：while (~scanf("%d%d", &a, &b))chatgpt给出的解释是：

在这段代码中，~ 被用来确保在 scanf 读取失败（返回 EOF）时，循环会退出。这是一种比较隐蔽的写法。通常情况下，开发者会直接用 while (scanf(...) != EOF) 来代替这种写法，会更加直观和易读。

 开发商购买土地问题

这是一个前缀和的应用问题

题目链接

44. 开发商购买土地 | 代码随想录

 题目可以这样理解的：

 本题可以使用 前缀和的思路来求解，先将 行方向，和 列方向的和求出来，这样可以方便知道 划分的两个区间的和。

先看代码：

#include <iostream>
#include <vector>
#include <climits>

using namespace std;
int main () {
    int n, m;
    cin >> n >> m;
    int sum = 0;
    vector<vector<int>> vec(n, vector<int>(m, 0)) ;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> vec[i][j];
            sum += vec[i][j];
        }
    }
    // 统计横向
    vector<int> horizontal(n, 0);
    for (int i = 0; i < n; i++) {
        for (int j = 0 ; j < m; j++) {
            horizontal[i] += vec[i][j];
        }
    }
    // 统计纵向
    vector<int> vertical(m , 0);
    for (int j = 0; j < m; j++) {
        for (int i = 0 ; i < n; i++) {
            vertical[j] += vec[i][j];
        }
    }
    int result = INT_MAX;
    int horizontalCut = 0;
    for (int i = 0 ; i < n; i++) {
        horizontalCut += horizontal[i];
        result = min(result, abs(sum - horizontalCut - horizontalCut));
    }
    int verticalCut = 0;
    for (int j = 0; j < m; j++) {
        verticalCut += vertical[j];
        result = min(result, abs(sum - verticalCut - verticalCut));
    }
    cout << result << endl;
}

 感觉很好理解，就是把上面问题从一维变成了二维

一维要定义的变量：presum（总和）,p[i]（前缀和数组）,a,b（上下界）

二维要定义的变量：sum（总和），vec[i][j]（前缀和数组）->horizontal[i]（水平前缀和数组）vertical[j]（垂直前缀和数组）horizontalCut （水平划分后上面值多大）verticalCut （垂直划分后左边值多大）

或者写成这样：

    int result = INT_MAX;
    int count = 0; // 统计遍历过的行
    for (int i = 0; i < n; i++) {
        for (int j = 0 ; j < m; j++) {
            count += vec[i][j];
            // 遍历到行末尾时候开始统计
            if (j == m - 1) result = min (result, abs(sum - count - count));

        }
    }

    count = 0; // 统计遍历过的列
    for (int j = 0; j < m; j++) {
        for (int i = 0 ; i < n; i++) {
            count += vec[i][j];
            // 遍历到列末尾的时候开始统计
            if (i == n - 1) result = min (result, abs(sum - count - count));
        }
    }

下面这个解法我很推荐

昨天没学习，今天预计还会发两篇博客，欢迎各位大佬评论区指出问题！