最大整除子集

f[i] 表示在输入数组 nums 升序排列的前提下,以 nums[i] 为最大整数的「整除子集」的大小。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
class Solution {
public:
vector<int> largestDivisibleSubset(vector<int>& nums) {
int n = nums.size();
vector<int> f(n + 1, 1);
sort(nums.begin(), nums.end());
int maxSize = 1;
int maxVal = f[0];
for (int i = 1; i < n; i++) {
for (int j = 0; j < i; j++) {
if (nums[i] % nums[j] == 0) {
f[i] = max(f[i], f[j] + 1);
}
}
if (f[i] > maxSize) {
maxSize = f[i];
maxVal = nums[i];
}
}
vector<int> res;
if (maxSize == 1) {
return {nums[0]};
}
for (int i = n - 1; i >= 0 && maxSize > 0; i--) {
if (f[i] == maxSize && maxVal % nums[i] == 0) {
res.push_back(nums[i]);
maxVal = nums[i];
maxSize--;
}
}
return res;
}
};