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

       문제 요약        

- 문제

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

프로그래머스

 

 

         아이디어        

한 상자에 담으려는 귤의 개수 k와 귤의 크기를 담은 배열 tangerine

         소스코드        

 

import java.util.*;
class Solution {
    static int loser = 0;
    static int cnt = 0;
    static Set<String> prevWords;
   
    public int[] solution(int n, String[] words) throws Exception {
        prevWords = new HashSet<>();
        prevWords.add(words[0]); // 첫번째 단어 무조건 포함
        for(int w = 1; w < words.length; w++) {
            String prev = words[w-1];
            if(prev.charAt(prev.length()-1) == words[w].charAt(0)) { // 이전 단어의 맨 마지막이랑 비교 
                if(prevWords.contains(words[w])) {
                    loser = w % n + 1;
                    cnt = (w + 1) / n + ((w+1) % n == 0? 0 : 1);
                    break;
                } else {
                    prevWords.add(words[w]);
                }
            } else {
                loser = w % n + 1;
                cnt = (w + 1) / n +  ((w+1) % n == 0? 0 : 1);
                break;
            }
        }
        if(loser == 0) {
            return new int[]{0, 0};
        }
        return new int[]{loser, cnt};
    }
}