Given an integer array
Consider the number of unique elements of
-
Change the array
nums such that the firstk elements ofnums contain the unique elements in the order they were present innums initially. The remaining elements ofnums are not important as well as the size ofnums . -
Return
k .
Custom Judge:
The judge will test your solution with the following code:
If all assertions pass, then your solution will be accepted.
Output: 2, nums = [1,2,_]
Explanation: Your function should return k = 2, with the first two elements of nums being 1 and 2 respectively.
It does not matter what you leave beyond the returned k (hence they are underscores).
Output: 5, nums = [0,1,2,3,4,_,_,_,_,_]
Explanation: Your function should return k = 5, with the first five elements of nums being 0, 1, 2, 3, and 4 respectively.
It does not matter what you leave beyond the returned k (hence they are underscores).
Constraints:
1 <= nums.length <= 3 * 104 -100 <= nums[i] <= 100 nums is sorted in non-decreasing order.
Contents
Solution 1 - Move distinct elements to the front of array
In this approach, we will move the distinct elements to the front of the array and ignore remaining.
Implementation steps:
-
We will create two variables
dupIndex to track the index ofnums array where to move distinct elements to, andprevElement to track current indexed element to check whether next coming elements are duplicates or not. -
We will initialize
dupIndex =1 , since element at 0th index will not have any elements before that, andprevElement = nums[0] . -
Then, we will loop through elements of
nums array from 1st index and check ifnums[i] is not same asprevElement , that means they are not duplicates, so move element at indexi todupIndex and incrementdupIndex , and update theprevElement with current value. If the current element is same asprevElement then continue the loop. -
Finally, return
dupIndex , it will be at the position where all distinct elements are moved to its left side.
Example: consider below input nums = [1,1,2,3]
-
As a first step, we will initialize
dupIndex = 1 andprevElement = nums[0] = 1 . -
Now, loop through
nums array from index 1. -
Element at 1st index = 1, is same as
prevElement , so we continue to next element. -
Element at 2nd index = 2, is not same as
prevElement (2 ≠ 1):-
So, we update
prevElement with current value,prevElement = 2 -
Move current indexed element to
dupIndex , so the resultant array looks like below: -
Now, increment
dupIndex to 2.
-
So, we update
-
Element at 3rd index = 3, is not same as
prevElement (3 ≠ 2):-
So, we update
prevElement with current value,prevElement = 3 -
Move current indexed element to
dupIndex , so the resultant array looks like below: -
Now, increment
dupIndex to 3.
-
So, we update
-
At the end, we will simply return the
dupIndex , this represents the the index where it left off, and that is also the length of distinct elements size.
Complexity Analysis:
Time complexity: Above code runs in O(n) time where
Space complexity: O(1).
Above implementations source code can be found at