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

 

 

문제 링크

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

프로그래머스

 

 

 

문제 풀이

1. n명의 사람이 번갈아가며 words의 단어를 말하며, words의 순서는 바꿀 수 없다.

2. n번째 사람 다음 순서는 1번째 사람이다.

3. 중복되는 단어는 허용하지 않는다.

4. 첫 알파벳이 이전 사람의 단어의 마지막 알파벳과 동일해야한다.

5. 끝말잇기에 실패한 사람의 번호와 차례를 return한다.

6. 실패한 사람이 없는 경우 [0, 0]을 return한다.

 

 

 

코드

import java.util.Set;
import java.util.HashSet;

class Solution {
    public int[] solution(int n, String[] words) {
        Set<String> set = new HashSet();
        char startChar;
        
        for(int i =1; i<words.length ;i++){
            if(set.contains(words[i]) || startChar !=  words[i].charAt(0)) 
                   return new int[]{i%n+1,i/n+1};
                   
            set.add(words[i]);
            startChar = words[i].charAt(words[i].length()-1);
        }
        return new int[]{0,0};
    }
}

 

 

 

실행 결과