[Programmers] 영어 끝말잇기

https://school.programmers.co.kr/learn/courses/30/lessons/12981

프로그래머스

문제 설명

Set을 활용한 간단한 구현 문제이다.

코드

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

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0};
        Set<String> set = new HashSet<>();
        set.add(words[0]);
        
        for (int order = 1; order < words.length; order++) {
            String currentWord = words[order];

            if (set.contains(currentWord)) {
                answer[0] = (order % n) + 1;
                answer[1] = (order / n) + 1;
                break;
            }
            
            String prevWord = words[order - 1];
            if (prevWord.charAt(prevWord.length() - 1)
                != (currentWord.charAt(0))) {
                answer[0] = (order % n) + 1;
                answer[1] = (order / n) + 1;
                break;
            }
            
            set.add(currentWord);
        }

        return answer;
    }
}