프로그래머스 lv2 영어 끝말잇기

문제 링크

영어 끝말잇기

풀이

• 이게 왜 lv2 문제지 싶을정도로 쉬운 문제였다.
• 그냥 단순하게 조건문으로 앞의 단어 끝글자와 뒤의 단어 첫글자가 같은지 확인하고, 그 다음에 이미 사용된 단어인지 확인만 해주면 된다.
• 그리고 리턴 값은 규칙을 조금만 찾아보면 알 수 있듯 i번째 / n + 1, i번째 / n + 1 으로 구해주면 된다.

import java.util.HashSet;
import java.util.Set;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0};

        Set<String> used = new HashSet<>();
        for (int i = 0; i < words.length; i++) {
            if ((i > 0 && words[i - 1].charAt(words[i - 1].length() - 1) != words[i].charAt(0)) || used.contains(words[i])) {
                return new int[]{(i % n) + 1, (i / n) + 1};
            }

            used.add(words[i]);
        }

        return answer;
    }
}