技術標籤：九章演算法面試題微軟面試字串java

描述

在物質位面“現實”中，有n+1個星球，分別為星球0，星球1，……，星球n。

每一個星球都有一個傳送門，通過傳送門可以直接到達目標星球而不經過其他的星球。

不過傳送門有兩個缺點。

第一，從星球i通過傳送門只能到達編號比i大，且與i的差不超過limit的星球。

第二，通過傳送門到達星球j，需要cost[j]個金幣。

現在，流浪劍客斯溫到達星球0後身上有m個金幣，請問他有多少種方法通過傳送門到達星球n？

• 1 <= n <= 50, 0 <= m <= 100, 1 <= limit <= 50, 0 <= cost[i] <= 100。
• 由於cost[0]沒有意義，題目保證cost[0] = 0。

線上評測地址

樣例1

比如 n = 15, 返回一個字串陣列：

輸入:
n = 1
m = 1, 
limit = 1
cost = [0, 1]
輸出:
1
解釋:
方案1：星球0→星球1

樣例2

輸入:
n = 1
m = 1
limit = 1
cost = [0,2]
輸出:
0
解釋:
無合法方案

演算法：dp

我們用dp[i][j]dp[i][j]代表從星球00出發到達星球ii後擁有jj個金幣的方案數。

• 設定初始狀態為在第0號星球，此時擁有m個幣。dp[0][m]=1dp[0][m]=1。
• 我們考慮dp[i][j]dp[i][j]的前繼狀態，可以發現，所有編號比i小，且差在limit之內的都能轉移過來，並且轉移過程消耗cost[i]cost[i]的幣，所以對它的前繼狀態的方案數累加。
• 可列出狀態轉移方程如下所示，

• 最後因為要求總方案數，對於sven在第nn號星球的所有剩餘金幣數量求和即可。答案

複雜度分析

• 時間複雜度O(n∗m∗limit)O(n∗m∗limit)
• 空間複雜度O(n∗m)O(n∗m)
  • 就是dpi所有的狀態數

public class Solution {
    /**
     * @param n: the max identifier of planet.
     * @param m: gold coins that Sven has.
     * @param limit: the max difference.
     * @param cost: the number of gold coins that reaching the planet j through the portal costs.
     * @return: return the number of ways he can reach the planet n through the portal.
     */
    public long getNumberOfWays(int n, int m, int limit, int[] cost) {
        // 
        long[][] dp = new long[n + 1][m + 1];
        for (int i = 0; i < m; i++) {
            dp[0][i] = 0;
        }
        dp[0][m] = 1;
        for (int i = 1; i <= n; i++) {
            for (int j = 0; j <= m; j++) {
                dp[i][j] = 0;
                for (int t = Math.max(0, i - limit); t <= i - 1; t++) {
                    if (j + cost[i] <= m) {
                        dp[i][j] += dp[t][j + cost[i]];
                    }
                }
            }
        }
        long ans = 0;
        for (int i = 0; i <= m; i++) {
            ans += dp[n][i];
        }
        return ans;
    }
}

 

更多題解參考