三角形最小路径和

三角形最短路径和

1.动态规划求解

给定一个三角形 triangle ,找出自顶向下的最小路径和。
每一步只能移动到下一行中相邻的结点上。相邻的结点 在这里指的是 下标 与 上一层结点下标 相同或者等于 上一层结点下标 + 1 的两个结点。也就是说,如果正位于当前行的下标 i ,那么下一步可以移动到下一行的下标 i 或 i + 1 。
题目链接

三角形最小路径和

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        for(int i=1;i<triangle.size();i++)
        {
            for(int j=0;j<triangle[i].size();j++)
            {
                if(j==0)
                triangle[i][j]+=triangle[i-1][j];
                else if(j==triangle[i].size()-1)
                triangle[i][j]+=triangle[i-1][j-1];
                else
                triangle[i][j]+=fmin(triangle[i-1][j-1],triangle[i-1][j]);
            }
        }
        
        int sub=triangle.size()-1;//获取最后一行的坐标
        int sum=triangle[sub][0];
        for(int i=0;i<triangle[sub].size();i++)
        {
            sum=fmin(sum,triangle[sub][i]);
        }
        return sum;
    }
};

2.不破坏原始空间写法

上面的解法,我们直接利用了数组的空间,我们可以自己开辟一个二维数组,即不破坏原始的空间

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        vector<vector<int>> ret(triangle.size(),vector<int>(triangle.size(),0));
        ret[0][0]=triangle[0][0];

        for(int i=1;i<triangle.size();i++)
        {
            for(int j=0;j<triangle[i].size();j++)
            {
                if(j==0)
                ret[i][j]=triangle[i][j]+ret[i-1][j];
                else if(j==triangle[i].size()-1)
                ret[i][j]=triangle[i][j]+ret[i-1][j-1];
                else
                ret[i][j]=triangle[i][j]+fmin(ret[i-1][j-1],ret[i-1][j]);
            }
        }
        
        int sub=triangle.size()-1;
        int sum=ret[sub][0];
        for(int i=0;i<triangle[sub].size();i++)
        {
            sum=fmin(sum,ret[sub][i]);
        }
        return sum;
    }
};

3.空间复杂度优化

第二种解法之中,空间复杂度为O(N^2),而又dp[i][j]只与dp[i-1][j]和dp[i-1][j-1]有关系,即当前行只与上一行有关系,因此可以定义一个一维数组重复使用,为了避免数据被覆盖的问题,我们从后往前遍历每一行

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {
        vector<int> ret(triangle.size(),0);
        ret[0]=triangle[0][0];
        for(int i=1;i<triangle.size();i++)
        {
            for(int j=triangle[i].size()-1;j>=0;j--)//避免被覆盖,从后往前进行遍历
            {
                if(j==0)
                ret[j]+=triangle[i][j];
                else if(j==triangle[i].size()-1)
                ret[j]=triangle[i][j]+ret[j-1];
                else
                ret[j]=triangle[i][j]+fmin(ret[j],ret[j-1]);
            }
        }

        int min=ret[0];
        for(int i=0;i<triangle.size();i++)
        {
            min=fmin(min,ret[i]);
        }
        return min;
    }
};

4.自低向上求解

上述的方法都是从上至下进行求解,因此最后还需要遍历一遍寻求最小值,如果从下至上求解,那么最后就只剩下一个最小值了

class Solution {
public:
    int minimumTotal(vector<vector<int>>& triangle) {

        for(int i=triangle.size()-2;i>=0;i--)//从倒数第二行开始
        {
            for(int j=triangle[i].size()-1;j>=0;j--)
            {
                triangle[i][j]+=fmin(triangle[i+1][j],triangle[i+1][j+1]);
            }
        }
        return triangle[0][0];
    }
};
上一篇:3.30刷题记录 Triangle


下一篇:动态规划:路径系列