这一题,我首先将nums2和nums1用vector<pair<int,int>>存储后排序,得到根据nums2[i]一个从小到大排序的数组。因为增长速度快的数据最后赋值为0,才能得到最小的和。
最大次数应该是调整n次,因为如果需要调整n+1次的话,那必定有重复的数据被赋值了两次0,那么前面的那一次就可以不用赋值为0,因为赋值一次0,其他的数局都会增加。
然后就是for循环调整次数从0到n-1(也可以1到n,那么下面代码就需要修改下标了),计算每一调整的和sum,如果找到一个sum小于等于x,那么就直接返回i+1。否则返回-1。代码如下:
class Solution {
public:
int minimumTime(vector<int>& nums1, vector<int>& nums2, int x) {
int size = nums1.size();
vector<pair<int,int>> v;
for(int i = 0 ; i < size; i++){
v.push_back({nums2[i],nums1[i]});
}
sort(v.begin(),v.end());
for(int i = 0; i < size; i++){
int sum = 0;
for(int j = 0; j < size; j++){
if(i == j) v[j].second = 0;
else v[j].second += v[j].first;
sum += v[j].second;
}
if(sum <= x) return i+1;
}
return -1;
}
};
但是只过了部分案例。
经对比题解发现,我猜是忽略了nums1的初始值,当遇到初始值比较大但是每一次都调整增长速度慢的数据不一定得到的是最小的调整次数。
题解如下:
class Solution {
public:
int minimumTime(vector<int>& nums1, vector<int>& nums2, int x) {
int n = nums1.size();
vector<int> dp(n + 1, 0);
vector<pair<int, int>> nums(n);
for (int i = 0; i < n; i++) {
nums[i] = {nums2[i], nums1[i]};
}
sort(nums.begin(), nums.end());
for (int j = 1; j <= n; j++) {
int b = nums[j - 1].first, a = nums[j - 1].second;
for (int i = j; i > 0; i--) {
dp[i] = max(dp[i], dp[i - 1] + i * b + a);
}
}
int s1 = accumulate(nums1.begin(), nums1.end(), 0);
int s2 = accumulate(nums2.begin(), nums2.end(), 0);
for (int i = 0; i <= n; i++) {
if (s2 * i + s1 - dp[i] <= x) {
return i;
}
}
return -1;
}
};