时间戳

首页 > 解决方案 > leetcode 1039 之 Python 解决方案 vs C++ 解决方案

python - leetcode 1039 之 Python 解决方案 vs C++ 解决方案

问题描述

力码 1039

我用python写了一段代码。

class Solution(object):
    def minScoreTriangulation(self, A):
        self.dp = [[-1]*len(A)]*len(A)
        ret = self.calc(A, 0, len(A)-1)
        return ret

    def calc(self, A, x, y):
        if math.ceil(y-x) < 2:
            return 0

        if self.dp[x][y] != -1:
            return self.dp[x][y]

        mn = sys.maxint
        for i in range(x+1, y):
            mn = min(mn, (self.calc(A, x, i) + self.calc(A, i, y) + A[x]*A[y]*A[i]))

        self.dp[x][y] = mn
        return mn

我还用相同的逻辑编写了一个 C++ 解决方案。

class Solution {
public:
    int minScoreTriangulation(vector<int>& A) {
        vector<int> row(A.size(), -1);
        vector<vector<int>> dp(A.size(), row);

        int ret =calc(dp, A, 0, A.size()-1);

        return ret;
    }

    int calc(vector<vector<int>>& dp, vector<int>& A, int x, int y)
    {
        if ((y-x) < 2)
            return 0;

        if(dp[x][y] != -1) return dp[x][y];

        int mn = INT_MAX;

        for( int i=x+1; i<y ; i++)
        {
            mn = min(mn, calc(dp, A, x, i)+calc(dp, A, i, y)+A[x]*A[y]*A[i]);
        }
        dp[x][y] =mn;
        return mn;
    }
};

python 解决方案给了我错误的答案。有人可以解释一下我的 python 解决方案有什么问题吗？

标签： pythonc++dynamic-programming

解决方案

问题是这一行：

self.dp = [[-1]*len(A)]*len(A)

在python中，你不能用这种方式来初始化一个二维数组，因为它会引用同一个[-1]*len(A)，改成：

self.dp = [[-1]*len(A) for i in range(len(A))]

它会正常工作。

推荐阅读