P12981. 영어 끝말잇기

문제

제출 코드

• 사용 알고리즘 : 자료구조

Set을 사용해서 언급된 단어를 저장하고, 매 차례마다 중복값을 말하는지와 이전 단어 끝글자로 시작하는지만 확인하면 되는 문제

import java.util.*;

class Solution {
    public int[] solution(int n, String[] words) {

        int answer[] = new int[2];
        Set<String> set = new HashSet<String>();
        char pre = words[0].charAt(words[0].length()-1);
        int len = words.length, idx=1;
        set.add(words[0]);
        answer[0]=1;
        for(int i=1; i<len; i++){
            if(set.contains(words[i]) || words[i].length()<=1 || pre!=words[i].charAt(0)){
                answer[0]++;
                answer[1] = i/n + 1;
                return answer;
            }else{
                answer[0] = (answer[0]+1)%n;
                set.add(words[i]);
                pre = words[i].charAt(words[i].length()-1);
            }
        }

        answer[0] = answer[1] = 0;
        return answer;
    }
}