DAY 24 LeetCode學習筆記

因為粉色的滿天星不會凋謝發表於2020-09-30

【313超級醜數】

前言

明天國慶加中秋,上了一天課,苦逼回到實驗室繼續學習,導師還拿了一百多份本科作業讓我評改,難受。

題目

官方題目

題解原始碼

這題和昨天的醜數2是一樣的方法,只不過昨天只有三個數,今天數不確定,採用了多指標和冬天規劃的方法。可參考醜數2部落格
Java原始碼

class Solution {
    public int nthSuperUglyNumber(int n, int[] primes) {
        int[] dp=new int[n];
        dp[0]=1;

        int k = primes.length;
        int [] index=new int[k];

        for (int i=1;i<n;++i){
            int min =Integer.MAX_VALUE;
            for(int j=0;j<k;++j){
                if (min>dp[index[j]]*primes[j]){
                    min=dp[index[j]]*primes[j];
                }
            }
            dp[i]=min;
            for(int j=0;j < k; ++j){
                if (min==dp[index[j]]*primes[j]){
                    index[j]++;
                }
            }
        }
        return dp[n-1];

        


    }
}

python


```python
class Solution:
    def nthSuperUglyNumber(self, n: int, primes: List[int]) -> int:
        dp=[1,]
        k=len(primes)
        index=[0]*k
        for i in range(1,n):
            minValue=float('inf')
            for j in range(k):
                if minValue>dp[index[j]]*primes[j]:
                    minValue=dp[index[j]]*primes[j]
            dp.append(minValue)
            for j in range(k):
                if minValue==dp[index[j]]*primes[j]:
                    index[j]+=1
        return dp[n-1]

相關文章