[프로그래머스/Lv.2] 영어 끝말잇기

1. Input

1) 끝말잇기 참가자의 수 n

2) 사람들이 순서대로 말한 단어를 담은 문자열 배열 words

 

2. Output

1) 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 담은 int 배열

2) 탈락자가 없다면 {0, 0} 반환

 

3. Constraint

1) 1번부터 번호 순서대로 한 사람씩 차례대로 단어를 말한다.

2) 마지막 사람이 단어를 말한 다음에는 다시 1번부터 말한다.

3) 앞사람이 말한 단어의 마지막 문자로 시작하는 단어를 말해야 한다.

4) 이전에 등장했던 단어는 사용할 수 없다.

5) 한 글자인 단어는 인정되지 않는다.

6) n은 2 이상 10 이하의 자연수

7) words의 길이는 n 이상 100 이하의 자연수

8) 단어의 길이는 2 이상 50 이하

9) 모든 단어는 알파벳 소문자로만 이루어져 있다.

10) 끝말잇기에 사용되는 단어의 뜻은 고려하지 않는다.

 

4. Example

Input: n=2, words={"hello", "one", "even", "never", "now", "world", "draw"} -> Output: {1, 3}

설명:

1번 "hello" -> 2번 "one" -> 1번 "even" -> 2번 "never" -> 1번 "now"에서 r로 시작하는 단어를 말한 것이 아니므로 1번이 탈락자고, 1번이 3번째로 말할 차례에서 탈락한 것이므로 {1, 3}을 반환한다.

 

5. Code

1) 첫 코드(2022/11/29)

import java.util.*;

ArrayList<String> list = new ArrayList<String>();
list.add(words[0]);
int[] answer = new int[2];
for(int i=1 ; i<words.length ; i++){
    if(list.contains(words[i]) || list.get(list.size()-1).charAt(
                                          list.get(list.size()-1).length()-1)
                                  ! =words[i].charAt(0)){
        answer[0] = (i+1)%n != 0 ? (i+1)%n : n;
        answer[1] = (i+1)%n != 0 ? (i+1)/n + 1 : (i+1)/n;
        break;
    }
    list.add(words[i]);
}
return answer;

2) 복잡한 코드를 정리한 후(2022/11/29)

import java.util.*;

HashSet list = new HashSet();
list.add(words[0]);
int[] answer = new int[2];
for(int i=1 ; i<words.length ; i++){
    if(list.contains(words[i]) || words[i-1].charAt(words[i-1].length()-1)
                                  != words[i].charAt(0)){
        answer[0] = i%n + 1;
        answer[1] = i/n + 1;
        break;
    }
    list.add(words[i]);
}
return answer;

  - 2번이 미세하게 더 빨랐다.