LeetCode 1289 下降路径最小和 II

给出一个二维数组,从每行选择一个数字,且相邻的两行不能选择同一列的数字,求选出来的所有数字最小和。很常规的一道动态规划题目,用一个数组dp[i][j]记录选择第i行第j列时的最小和,我是从上到下选择,每次选择不能选择和前一行同列的数。

class Solution {
public:
    int dp[205][205];
    struct node
    {
        int val,pos;
        bool operator <(const node& rhs) const
        {
            return val<rhs.val;
        }
    };
    int minFallingPathSum(vector<vector<int>>& arr) {
        int n=arr.size(),m;
        if(n!=0) m=arr[0].size();
        for(int i=0;i<m;i++)
        dp[0][i]=arr[0][i];
        node tmp[205];
        for(int i=1;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                tmp[j].pos=j;
                tmp[j].val=dp[i-1][j];
            }
            sort(tmp,tmp+m);
            for(int j=0;j<m;j++)
            {
                if(tmp[0].pos!=j)
                    dp[i][j]=arr[i][j]+tmp[0].val;
                else
                    dp[i][j]=arr[i][j]+tmp[1].val;
            }
        }
        int ans=0x3f3f3f3f;
        for(int i=0;i<m;i++)
            ans=min(ans,dp[n-1][i]);
        return ans;
    }
};
posted @ 2020-05-09 12:09  South1999  阅读(95)  评论(0编辑  收藏  举报