思路
周赛时候卡了好久,考虑了向前贪心替换和向后替换,都不行,周赛结束了看大佬代码才明白过来。
对于步长为k,会产生k个数组,0,k,2k;1,1+k,1+2k…,这要这些数组各自满足递增即可。
因为可以改为任意数,所以不需要管改为什么数,只需要记要改哪些即可,而贪心的想法即是将最长递增子序列之外的改掉,所以需要改的操作数为分组数组的长度减去该数组最长递增子序列的长度,答案即为这些操作数之和。
而对于一个数组,其最长递增子序列的求法为贪心dp,见力扣第300题,用一个dp数组存储上升最慢的子序列,遇到新的数,如果比尾部大,加入数组;如果比尾部小,在该数组中二分查找到对应位置进行替换,遍历完后该dp数组长度即为最长递增子序列长度。
代码
class Solution {
public:
int LIS(vector<int>& arr, int st, int k) {
vector<int> kson;
for(int i = st; i < arr.size(); i+=k) {
kson.push_back(arr[i]);
}
int n = kson.size();
vector<int> dp(n+1, 0);
dp[0] = kson[0];
int length = 1;
for(int i = 1; i < n; i++) {
auto it = upper_bound(dp.begin(), dp.begin() + length, kson[i]) - dp.begin();
if(it == length) {
dp[length] = kson[i];
length++;
}
else dp[it] = kson[i];
}
return n - length;
}
int kIncreasing(vector<int>& arr, int k) {
int ret = 0;
for (int i = 0; i < k; i++) {
ret += LIS(arr, i, k);
}
return ret;
}
};