[프로그래머스 - 자바(JAVA)] 39 : 영어 끝말잇기

목차

영어 끝말잇기

---

영어 끝말잇기 : Lv.2

▼ 문제

출처 : 프로그래머스 코딩테스트 연습 > Summer/Winter Coding(~2018) > 영어 끝말잇기

▼ 내 풀이

• 각 단어마다 몇번 사람이 말했는지에 대한 정보를 가지고 있어야하므로 map을 사용했다.
• 중간에 break를 빼고 코드를 작성해서 많이 헤맸다.

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = new int[2];
        int count = 1; 
        Map<String, Integer> map = new HashMap<>();
        map.put(words[0], 0);
        
        for(int i = 1; i < words.length; i++) {
            if(map.containsKey(words[i]) || words[i].charAt(0) != words[i-1].charAt(words[i-1].length()-1)) {
                int indx = i % n + 1 ; 
                answer[0] = indx;
                answer[1] = count/n + 1;
                break;
            } else {
                map.put(words[i], i % n);
                count++;
            }
        }
        if(count == words.length) {
            answer[0] = 0;
            answer[1] = 0;
        }
        

        return answer;
    }
}

---