-
Notifications
You must be signed in to change notification settings - Fork 2
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
O(nlogn) time and O(n) space using sorting.
- Loading branch information
Showing
1 changed file
with
45 additions
and
0 deletions.
There are no files selected for viewing
45 changes: 45 additions & 0 deletions
45
...h That Maximum Difference Is K/2294. Partition Array Such That Maximum Difference Is K.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,45 @@ | ||
""" | ||
You are given an integer array nums and an integer k. You may partition nums into one or more subsequences such that each element in nums appears in exactly one of the subsequences. | ||
Return the minimum number of subsequences needed such that the difference between the maximum and minimum values in each subsequence is at most k. | ||
A subsequence is a sequence that can be derived from another sequence by deleting some or no elements without changing the order of the remaining elements. | ||
Example 1: | ||
Input: nums = [3,6,1,2,5], k = 2 | ||
Output: 2 | ||
Explanation: | ||
We can partition nums into the two subsequences [3,1,2] and [6,5]. | ||
The difference between the maximum and minimum value in the first subsequence is 3 - 1 = 2. | ||
The difference between the maximum and minimum value in the second subsequence is 6 - 5 = 1. | ||
Since two subsequences were created, we return 2. It can be shown that 2 is the minimum number of subsequences needed. | ||
Example 2: | ||
Input: nums = [1,2,3], k = 1 | ||
Output: 2 | ||
Explanation: | ||
We can partition nums into the two subsequences [1,2] and [3]. | ||
The difference between the maximum and minimum value in the first subsequence is 2 - 1 = 1. | ||
The difference between the maximum and minimum value in the second subsequence is 3 - 3 = 0. | ||
Since two subsequences were created, we return 2. Note that another optimal solution is to partition nums into the two subsequences [1] and [2,3]. | ||
Example 3: | ||
Input: nums = [2,2,4,5], k = 0 | ||
Output: 3 | ||
Explanation: | ||
We can partition nums into the three subsequences [2,2], [4], and [5]. | ||
The difference between the maximum and minimum value in the first subsequences is 2 - 2 = 0. | ||
The difference between the maximum and minimum value in the second subsequences is 4 - 4 = 0. | ||
The difference between the maximum and minimum value in the third subsequences is 5 - 5 = 0. | ||
Since three subsequences were created, we return 3. It can be shown that 3 is the minimum number of subsequences needed. | ||
""" | ||
class Solution: | ||
def partitionArray(self, nums: List[int], k: int) -> int: | ||
nums.sort() | ||
min_val,result = -sys.maxsize-1,0 | ||
for num in nums: | ||
if num > min_val + k: | ||
min_val = num | ||
result += 1 | ||
return result |