欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

[LeetCode]313. Super Ugly Number

程序员文章站 2022-03-07 18:09:18
...

[LeetCode]313. Super Ugly Number

题目描述

[LeetCode]313. Super Ugly Number

思路

动态规划
保持结果序列递增,需要确保每次放入结果序列的数是当前乘积最小,当前的乘积是由产生序列的数和素数数组里的数乘积得到的,因此需要一个状态数组来保存对于素数数组里的每个数,乘到了数组里的第几个数,最终生成序列

代码

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        int primesNum = primes.size();
        vector<int> uglyArr(n, INT_MAX), indexArr(primesNum, 0);
        uglyArr[0] = 1;
        for (int i = 1; i < n; ++i) {
            // 更新序列中的数
            for (int index = 0; index < primesNum; index++) uglyArr[i] = min(uglyArr[i], uglyArr[indexArr[index]] * primes[index]);
            // 序列中的数更新后,更新index数组
            for (int index = 0; index < primesNum; index++) indexArr[index] += (uglyArr[i] == uglyArr[indexArr[index]] * primes[index]);
        }
        return uglyArr[n - 1];
    }
};

int main() {
    vector<int> primes = { 2,7,13,19 };
    Solution s;
    cout << s.nthSuperUglyNumber(11, primes) << endl;

    system("pause");
    return 0;
}
相关标签: leetcode