[프로그래머스 12981] Level2 영어 끝말잇기(Java)

문제 URL

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

프로그래머스

 

문제풀이 Key Point

• 이전에 말한 단어의 끝 알파벳과 현재 말하는 단어의 앞 알파벳이 일치해야한다.
• 이전까지 말한 단어를 중복해서 말하면 안된다.

 

Java 문제풀이

import java.util.Map;
import java.util.HashMap;
import java.util.Set;
import java.util.HashSet;
 
class Solution {
    public int[] solution(int n, String[] words) {
        int[] answer = new int[2];
 
        Map<Integer, Integer> players = new HashMap<>();
        Set<String> wordGroup = new HashSet<>();
        
        for (int i = 0; i < n; i++) {
            players.put(i + 1, 0);
        }
        
        boolean prevWord = false;
        String lastAlphabet = "";
        String currentAlphabet = "";
        int turn = 1;
        for (String word : words) {
            players.put(turn, players.get(turn) + 1);
            
            currentAlphabet = word.substring(0, 1);
            if (prevWord) {
                if (!lastAlphabet.equals(currentAlphabet)) {
                    return new int[]{turn, players.get(turn)};
                }
            }
            
            if (wordGroup.contains(word)) {
                return new int[]{turn, players.get(turn)};
            }
            wordGroup.add(word);
            lastAlphabet = word.substring(word.length() - 1);
            prevWord = true;
            turn++;
            
            if (turn > n) {
                turn = 1;
            }
        }
 
        return answer;
    }
}