영어 끝말잇기 (Summer/Winter Coding(~2018))

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

프로그래머스

 

풀이

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {
        LinkedList<String> list = new LinkedList<>();

        for (int i = 0; i < words.length; i += n) {
            for (int j = 0; j < n; j++) {
                String str = words[i + j];

                if (list.size() > 0) {
                    String preStr = list.peekLast();
                    if (preStr.charAt(preStr.length() - 1) != str.charAt(0)) {
                        return new int[]{j + 1, i / n + 1};
                    }
                }

                if (list.contains(str)) {
                    return new int[]{j + 1, i / n + 1};
                }

                list.add(str);
            }
        }

        return new int[]{0, 0};
    }
}