TIL (2024.06.15)

문제 1️⃣ (🔗 리트 코드 _ 좋은 짝의 수)

💡 IDEA : 배열 안의 각수의 개수를 HashMap 에 저장한 후 전체를 돌면서 조합의 수를 더한다. 이 때 조합은 n * (n-1) / 2 로 계산 가능하다.

import java.util.Arrays;
import java.util.HashMap;
import java.util.Set;

class Solution {
    public int numIdenticalPairs(int[] nums) {

        // (1) 배열 속 전체 값 세기
        HashMap<Integer, Integer> cnt = new HashMap<>();

        int val = 0;

        for (int num : nums) {
            if (cnt.get(num) == null) {
                cnt.put(num, 1);
            }
            else {
                val = cnt.get(num);
                cnt.put(num, ++val);
            }
        }
        
        // (2) 조합 계산
        Set<Integer> keySet = cnt.keySet();

        int answer = 0;
        int n = 0;
        for (Integer key : keySet) {
            n = cnt.get(key);
            answer += n*(n-1)/2;
        }
        
        return answer;
    }
}

코드 개선 사항 ① :  HashMap 값 추가

➡️ 기존 코드에서는 get 매서드를 통해 HashMap 안에 해당 값이 있는지 확인하고, 있다면 해당 값을 가져와서 1을 더한 뒤 넣고, 없다면 바로 1을 넣었다. 하지만 getOrDefault 매서드를 활용하면 기존 값이 있다면 해당 값을 가져오고 아니면 지정한 값을 가져오도록 할 수 있다.

// 개선 전

int val = 0;

for (int num : nums) {
	if (cnt.get(num) == null) {
    cnt.put(num, 1);
 }
else {
	val = cnt.get(num);
	cnt.put(num, ++val);
}

// 개선 후

for (int num : nums) {
	cnt.put(num, cnt.getOrDefault(num, 0)+1);
}

코드 개선 사항 ② : HashMap 값 하나씩 꺼내기

➡️ 기존 코드에서는 keySet 매서드를 통해 HashMap 속 key 값들을 불러온 후 for 문을 통해서 key 값을 바탕으로 값을 불러왔다면, values 매서드를 통해 바로 value 들에 접근한다.

// 개선 전
Set<Integer> keySet = cnt.keySet();

int answer = 0;
int n = 0;

for (Integer key : keySet) {
    n = cnt.get(key);
    answer += n*(n-1)/2;
}

// 개선 후

int answer = 0;

for (int count : cnt.values()) {
	answer += count * (count - 1) / 2;
}

최종 코드

class Solution {
    public int numIdenticalPairs(int[] nums) {

        // (1) 배열 속 전체 값 세기
        HashMap<Integer, Integer> cnt = new HashMap<>();

        for (int num : nums) {
	    cnt.put(num, cnt.getOrDefault(num, 0)+1);
        }
        
        // (2) 조합 계산
        int answer = 0;

        for (int n : cnt.values()) {
            answer += n*(n-1)/2;
        }

        return answer;
    }
}

문제 2️⃣ (🔗 리트 코드  _ XOR 연산을 통한 원래 배열 찾기)

💡 IDEA : Java 에서 지원하는 XOR 연산인 ^ 기호를 통해 해결하였다.

class Solution {
    public int[] findArray(int[] pref) {
        int[] answer = new int[pref.length];

        answer[0] = pref[0];

        for (int i=1; i<pref.length; i++) {
            answer[i] = pref[i-1] ^ pref[i];
        }

        return answer;
    }
}