1471. The k Strongest Values in an Array LeetCode Solution
In this guide, you will get 1471. The k Strongest Values in an Array LeetCode Solution with the best time and space complexity. The solution to The k Strongest Values in an Array problem is provided in various programming languages like C++, Java, and Python. This will be helpful for you if you are preparing for placements, hackathons, interviews, or practice purposes. The solutions provided here are very easy to follow and include detailed explanations.
Table of Contents
- Problem Statement
- Complexity Analysis
- The k Strongest Values in an Array solution in C++
- The k Strongest Values in an Array solution in Java
- The k Strongest Values in an Array solution in Python
- Additional Resources
Problem Statement of The k Strongest Values in an Array
Given an array of integers arr and an integer k.
A value arr[i] is said to be stronger than a value arr[j] if |arr[i] – m| > |arr[j] – m| where m is the median of the array.
If |arr[i] – m| == |arr[j] – m|, then arr[i] is said to be stronger than arr[j] if arr[i] > arr[j].
Return a list of the strongest k values in the array. return the answer in any arbitrary order.
Median is the middle value in an ordered integer list. More formally, if the length of the list is n, the median is the element in position ((n – 1) / 2) in the sorted list (0-indexed).
For arr = [6, -3, 7, 2, 11], n = 5 and the median is obtained by sorting the array arr = [-3, 2, 6, 7, 11] and the median is arr[m] where m = ((5 – 1) / 2) = 2. The median is 6.
For arr = [-7, 22, 17, 3], n = 4 and the median is obtained by sorting the array arr = [-7, 3, 17, 22] and the median is arr[m] where m = ((4 – 1) / 2) = 1. The median is 3.
Example 1:
Input: arr = [1,2,3,4,5], k = 2
Output: [5,1]
Explanation: Median is 3, the elements of the array sorted by the strongest are [5,1,4,2,3]. The strongest 2 elements are [5, 1]. [1, 5] is also accepted answer.
Please note that although |5 – 3| == |1 – 3| but 5 is stronger than 1 because 5 > 1.
Example 2:
Input: arr = [1,1,3,5,5], k = 2
Output: [5,5]
Explanation: Median is 3, the elements of the array sorted by the strongest are [5,5,1,1,3]. The strongest 2 elements are [5, 5].
Example 3:
Input: arr = [6,7,11,7,6,8], k = 5
Output: [11,8,6,6,7]
Explanation: Median is 7, the elements of the array sorted by the strongest are [11,8,6,6,7,7].
Any permutation of [11,8,6,6,7] is accepted.
Constraints:
1 <= arr.length <= 105
-105 <= arr[i] <= 105
1 <= k <= arr.length
Complexity Analysis
- Time Complexity: O(\texttt{sort})
- Space Complexity: O(k)
1471. The k Strongest Values in an Array LeetCode Solution in C++
class Solution {
public:
vector<int> getStrongest(vector<int>& arr, int k) {
ranges::sort(arr);
const int n = arr.size();
const int median = arr[(n - 1) / 2];
vector<int> ans;
for (int l = 0, r = n - 1; k > 0; --k)
if (median - arr[l] > arr[r] - median)
ans.push_back(arr[l++]);
else
ans.push_back(arr[r--]);
return ans;
}
};
/* code provided by PROGIEZ */
1471. The k Strongest Values in an Array LeetCode Solution in Java
class Solution {
public int[] getStrongest(int[] arr, int k) {
Arrays.sort(arr);
final int n = arr.length;
final int median = arr[(n - 1) / 2];
int[] ans = new int[k];
for (int i = 0, l = 0, r = n - 1; i < k; ++i)
if (median - arr[l] > arr[r] - median)
ans[i] = arr[l++];
else
ans[i] = arr[r--];
return ans;
}
}
// code provided by PROGIEZ
1471. The k Strongest Values in an Array LeetCode Solution in Python
class Solution:
def getStrongest(self, arr: list[int], k: int) -> list[int]:
arr.sort()
ans = []
median = arr[(len(arr) - 1) // 2]
l = 0
r = len(arr) - 1
for _ in range(k):
if median - arr[l] > arr[r] - median:
ans.append(arr[l])
l -= 1
else:
ans.append(arr[r])
r += 1
return ans
# code by PROGIEZ
Additional Resources
- Explore all LeetCode problem solutions at Progiez here
- Explore all problems on LeetCode website here
Happy Coding! Keep following PROGIEZ for more updates and solutions.