[프로그래머스][summer/winter coding] 영어 끝말잇기

2020. 4. 7. 13:17프로그래머스/LEVEL 2

import java.util.*;

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

        List<String> passWord = new ArrayList<>();
        passWord.add(words[0]);
        
        for(int i=1; i<words.length; i++){
            if(!passWord.contains(words[i])
               && words[i].charAt(0)==words[i-1].charAt(words[i-1].length()-1)
              ){
                passWord.add(words[i]);
            }else{
                if(passWord.size()%n ==0){
                answer[0] = 1;
                answer[1] = passWord.size()/n +1;
                    break;
                }else{
                    answer[0] = passWord.size()%n +1;
                    answer[1] = passWord.size()/n +1;
                    break;
                }
            }
        }
        
        return answer;
    }
}