程序员

leetcode 746--DP

2018-12-20  本文已影响5人  Ariana不会哭
图片.png
int minCostClimbingStairs(vector<int>& cost) {
        if(cost.size()==0)
            return 0;
        int n=cost.size();
        vector<int>dp(n,0);
        if(n==2)
            return min(cost[0],cost[1]);
        if(n==1)
            return dp[0];
        dp[0]=cost[0];dp[1]=cost[1];
        for(int i=2;i<n;i++){
            dp[i]=min(dp[i-1],dp[i-2])+cost[i];
        }
        return min(dp[n-1],dp[n-2]);
    }

Java:

//faster
    public int minCostClimbingStairs(int[] cost) {
        int n = cost.length;
        int[] dp = new int[n];
        dp[0] = cost[0];
        dp[1] = cost[1];
        for (int i = 2; i < n; i++) {
            dp[i] = cost[i] + Math.min(dp[i - 1], dp[i - 2]);
        }
        return Math.min(dp[n - 1], dp[n - 2]);
    }
上一篇 下一篇

猜你喜欢

热点阅读