[LeetCode]313. Super Ugly Number
程序员文章站
2022-03-07 18:09:18
...
[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;
}
上一篇: HTML
下一篇: 使用table标签制作个人简历