矩阵的最小路径和（空间压缩）

https://www.nowcoder.com/practice/2fb62a4500af4f4ba5686c891eaad4a9tpId=101&tqId=33254&tPage=1&rp=1&ru=/ta/programmer-code-interview-guide&qru=/ta/programmer-code-interview-guide/question-ranking

题目描述

给定一个 n * m 的矩阵 a，从左上角开始每次只能向右或者向下走，最后到达右下角的位置，路径上所有的数字累加起来就是路径和，输出所有的路径中最小的路径和。

输入描述:

第一行输入两个整数 n 和 m，表示矩阵的大小。

接下来 n 行每行 m 个整数表示矩阵。

输出描述:

输出一个整数表示答案。

示例1

输入

4 4
1 3 5 9
8 1 3 4
5 0 6 1
8 8 4 0

输出

12

备注:

1≤n,m≤2000  0≤aij​≤100

//方法1：O(mn)  O(1)
//直接在改变data数据，使用它充当dp数组，开销小，但是把data数组破坏了
#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    vector<vector<int>> data(n,vector<int>(m,0));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>data[i][j];
        }
    }
    
    for(int i=1;i<n;i++){
        data[i][0]=data[i-1][0]+data[i][0];
    }
    
    for(int j=1;j<m;j++){
        data[0][j]=data[0][j-1]+data[0][j];
    }
    
    for(int i=1;i<n;i++){
        for(int j=1;j<m;j++){
            data[i][j]=min(data[i-1][j],data[i][j-1])+data[i][j];
        }
    }
    
    cout<<data[n-1][m-1]<<endl;
    return 0;
}

//方法2：O(mn)  O(m)
//保留原数据
#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    vector<vector<int>> data(n,vector<int>(m,0));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>data[i][j];
        }
    }
    
    vector<int> dp(m,0);
    dp[0]=data[0][0];
    for(int j=1;j<m;j++){
        dp[j]=dp[j-1]+data[0][j];
    }
    
    for(int i=1;i<n;i++){
        dp[0]=dp[0]+data[i][0];
        for(int j=1;j<m;j++){
            dp[j]=min(dp[j],dp[j-1])+data[i][j];
        }
    }
    
    cout<<dp[m-1]<<endl;
    return 0;
}

//方法3：对方法2的空间进一步优化 
//O(mn)  O(min(m,n))
//保留原数据
#include<bits/stdc++.h>
using namespace std;
int main(){
    int n,m;
    cin>>n>>m;
    vector<vector<int>> data(n,vector<int>(m,0));
    for(int i=0;i<n;i++){
        for(int j=0;j<m;j++){
            cin>>data[i][j];
        }
    }
    int more=(n>m) ? n : m;
    int less=(n>m) ? m : n;
    int flag=(n==more) ? 1 : 0; //flag==1,列小
    vector<int> dp(less,0);
    dp[0]=data[0][0];
    for(int i=1;i<less;i++){
        dp[i]=dp[i-1]+ (flag==1 ? data[0][i] : data[i][0]);
    }
    
    for(int i=1;i<more;i++){
        dp[0]=dp[0]+( flag==1 ? data[i][0] : data[0][i]);
        for(int j=1;j<less;j++){
            dp[j]=min(dp[j],dp[j-1])+ (flag==1 ? data[i][j] : data[j][i]);
        }
    }
    
    cout<<dp[less-1]<<endl;
    return 0;
}