⬛ [Programmers] [Java] [Level2] 영어 끝말잇기

문제

https://school.programmers.co.kr/learn/courses/30/lessons/12981?language=java

프로그래머스

풀이

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0};
        Map<String, Integer> map = new HashMap<>();

        for(int i=0; i<words.length; i++){
            if(i!=0){
                String prev = words[i-1];
                String curr = words[i];

                char last = prev.charAt(prev.length() - 1);
                char first = curr.charAt(0);

                if(map.containsKey(curr) || last != first){
                    answer[0] = (i % n) + 1;
                    answer[1] = (i / n) + 1;

                    return answer;
                }
            }
            map.put(words[i], 1);
        }
        return answer;
    }
}

이 문제는 Map 자료구조를 이용하여 풀 수 있는 문제이다. 처음에는 스택에 모든 값을 넣고 일일이 비교하려고 하였으나 효율적이지 않고 중복체크를 해야하기 때문에 Map을 이용하였다. 

 

1. 이전 단어를 prev, 현재 단어를 curr에 저장한다. 

2. 이전 단어의 마지막 문자를 last, 현재 단어의 첫번째 문자를 first에 저장

3. 만약 현재 단어가 map에 있을 경우 또는 마지막 문자와 현재의 첫번째 문자가 같지 않을 경우 탈락이므로 return 해준다. 

4. 그렇지 않을 경우 map에 현재의 단어를 put하고 다음으로 넘어간다. 

 

최종적으로 모든 words를 돌았을 때 실패하지 않으면 처음 answer를 초기화 했던 [0,0]이 반환되고 중간에 실패할 경우 answer가 반환된다. 

 

map말고 set이나 다른 자료구조를 사용하여 풀이를 할 수도 있으므로 참고하기 바란다.