leetcode - 26.Remove Duplicates from Sorted Array

들어가기 앞서, 위의 코딩테스트는 python을 위주로 진행하였지만, java를 사용하여 다시금 코딩테스트 문제들을 풀어보려한다.

주제

array 중복 해결

문제

https://leetcode.com/problems/remove-duplicates-from-sorted-array

array의 중복 요소를 지우고, array의 사이즈를 출력하라.

입출력 예시

예시1

입력: nums = [1, 1, 2] 출력: 2, nums = [1, 2, _]

예시2

입력: nums = [0,0,1,1,1,2,2,3,3,4] 출력: 5, nums = [0,1,2,3,4,_,_,_,_,_]

조건

• 1 <= nums.length <= $3 * 10^4$
• -100 <= nums[i] <= 100
• nums내림차순이 아닌 순서 로 정렬됩니다 .

문제풀이

set 이용

1. set 초기화를 진행한다. LinkedHashSet을 사용한 이유는 순서가 보장되기 때문
2. set에 있는 데이터를 nums에 덮어쓴다.
3. set의 크기를 리턴한다.

import java.util.Set;
import java.util.Iterator;
import java.util.LinkedHashSet;

class Solution {
    public int removeDuplicates(int[] nums) {
        Set<Integer> set = new LinkedHashSet<>();
        for (int num : nums) {
            set.add(num);
        }

        Iterator<Integer> iterator = set.iterator();
        for (int i = 0; i < nums.length; i++) {
            if (iterator.hasNext()) {
                nums[i] = iterator.next();
            }
        }

        return set.size();
    }
}

반복문

• 중요 키워드: n번째 요소와 n - 1를 비교하는 것

1. index 같은 변수를 선언한다.
2. 1부터 nums의 길이 만큼 반복할 때까지 반복문을 돈다.
3. 만약, nums의 n번째 요소와 n - 1번째 요소가 다르다면 nums[index]에 n번째 요소를 덮어쓴 후, index 변수를 증감한다.
4. index 변수를 리턴한다.

class Solution {
    public int removeDuplicates(int[] nums) {
        int index = 1;

        for (int i = 1; i < nums.length; i++) {
            if (nums[i - 1] != nums[i]) {
                nums[index] = nums[i];
                index++;
            }
        }

        return index;
    }
}