[LeetCode] Maximum Beauty of an Array After Applying Operation

2779. Maximum Beauty of an Array After Applying Operation

You are given a 0-indexed array nums and a non-negative integer k.

In one operation, you can do the following:

  • Choose an index i that hasn’t been chosen before from the range [0, nums.length - 1].
  • Replace nums[i] with any integer from the range [nums[i] - k, nums[i] + k].

The beauty of the array is the length of the longest subsequence consisting of equal elements.

Return the maximum possible beauty of the array nums after applying the operation any number of times.

Note that you can apply the operation to each index only once.

A subsequence of an array is a new array generated from the original array by deleting some elements (possibly none) without changing the order of the remaining elements.

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
public:
int maximumBeauty(vector<int>& A, int k) {
sort(begin(A), end(A));
int l = 0, r = 0, n = A.size(), res = 0;
while(r < n) {
while(r < n and A[l] + 2 * k >= A[r]) {
res = max(res, r - l + 1);
r += 1;
}
l += 1;
}
return res;
}
};
Author: Song Hayoung
Link: https://songhayoung.github.io/2023/07/16/PS/LeetCode/maximum-beauty-of-an-array-after-applying-operation/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.