[프로그래머스 연습문제/JAVA] 영어 끝말잇기

문제

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

프로그래머스

 

 

import java.util.*;

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

        int[] answer = {0, 0};
        int cnt = 1;
        int people = 0;

        List<String> wordList = new ArrayList<>();

        char last = words[0].charAt(0);

        for(int i=0; i< words.length; i++){
            if(wordList.contains(words[i]) || last != words[i].charAt(0)){
                return new int[]{people+1, cnt};
            }else{
                wordList.add(words[i]);
                last = words[i].charAt(words[i].length()-1);
            }

            if((i+1) % n == 0){
                people = 0;
                cnt ++;
            }else{
                people++;
            }

        }

        return answer;
    }
}