[프로그래머스] LEVEL2 영어 끝말잇기 JAVA

영어 끝말잇기 JAVA

문제링크

풀이

Set을 활용하여 중복된 단어를 확인한다
charAt을 활용하여 단어의 마지막 글자와 다음 사람의 첫글자를 비교한다.

코드

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0}; // 번호, 차례
        
        Set<String> set = new HashSet<>();
        
        for (int i = 0; i < words.length; i++){
            int j = i + 1;
            String curWord = words[i];
            if(set.contains(curWord) 
               || (i > 0 && words[i-1].charAt(words[i-1].length() - 1) != curWord.charAt(0))) {
                answer[0] = i % n + 1;
                answer[1] = i / n + 1;
                break;
            } else {
                set.add(curWord);
            }
        }

        return answer;
    }
}