金字塔数组从上到下打印最小权值和---------实现空间复杂度O(N)

版权声明：转载请注明出处即可 https://blog.csdn.net/qq_35170212/article/details/81252444

题目:

金字塔数组从上到下打印最短路径和
实现空间复杂度O(N)

手撕:

#include "mode.h"
using namespace std;
int minimumTotal(vector<vector<int>>& triangle) {
    int res=0;
    if(triangle.size()&&triangle[0].size()){
        int col=triangle[triangle.size()-1].size();
        int row=triangle.size();
        vector<int> vr(col);
        for(int i=0;i<col;++i){
            vr[i]=triangle[row-1][i];
        }
        int curcol=col-1;
        for(int i=row-2;i>=0;--i){
            for(int j=0;j<curcol;++j){
                vr[j]=min(vr[j],vr[j+1])+triangle[i][j];
            }
            --curcol;
        }         
        res=vr[0];
    }
    return res;
}

测试:

int main(){
    int n=0;
    while(cin>>n){        
        vector<vector<int>>vr;
        for(int i=0;i<n;i++){
            vector<int> t;
            for(int j=n-i-1;j<n;j++){
                int tt=0;
                cin>>tt;
                t.push_back(tt);
            }
            vr.push_back(t);
        }
        cout<<minimumTotal(vr)<<endl;
    }
    return 0;
}

扩展:

不是金字塔,是矩形左上到左下(传送门)