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

문제 풀이 방식

• word를 저장할 list를 두고, 여기에 이미 단어가 포함되어있거나 끝말잇기가 불가능한경우 answer 값 리턴. 아닌 경우 list에 저장.
• HashMap을 사용하는 풀이도 확인하기

문제 풀이 (Java)

import java.util.List;
import java.util.ArrayList;

class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = {0, 0};
        List<String> list = new ArrayList<>();

        for(int i = 0; i < words.length; i++){
            if (i != 0){
                String last = words[i - 1];
                String now = words[i];

                if (!list.contains(now) && 
                   last.charAt(last.length() - 1) == now.charAt(0)){
                    list.add(now);
                }else{
                    answer[0] = i % n + 1;
                    answer[1] = i / n + 1;
                    break;
                }
            }else{
                list.add(words[i]);
            }

        }

        return answer;
    }
}