1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.
다음은 3명이 끝말잇기를 하는 상황을 나타냅니다.
tank → kick → know → wheel → land → dream → mother → robot → tank
위 끝말잇기는 다음과 같이 진행됩니다.
끝말잇기를 계속 진행해 나가다 보면, 3번 사람이 자신의 세 번째 차례에 말한 tank 라는 단어는 이전에 등장했던 단어이므로 탈락하게 됩니다.
사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.
n | words | result |
---|---|---|
3 | ["tank", "kick", "know", "wheel", "land", "dream", "mother", "robot", "tank"] | [3,3] |
5 | ["hello", "observe", "effect", "take", "either", "recognize", "encourage", "ensure", "establish", "hang", "gather", "refer", "reference", "estimate", "executive"] | [0,0] |
2 | ["hello", "one", "even", "never", "now", "world", "draw"] | [1,3] |
입출력 예 #13명의 사람이 끝말잇기에 참여하고 있습니다.
tank
와 같은 순서로 말을 하게 되며, 3번 사람이 자신의 세 번째 차례에 말한 tank
라는 단어가 1번 사람이 자신의 첫 번째 차례에 말한 tank
와 같으므로 3번 사람이 자신의 세 번째 차례로 말을 할 때 처음 탈락자가 나오게 됩니다.
주어진 영어 단어의 첫말과 끝말이 일치하지 않는 경우와 더불어 중복되는 단어를 말하는 경우도 신경을 써야하는 문제이다.
나는 이러한 제한사항을 List
자료구조에 영어 단어를 담고 단어별 끝 인덱스와 첫 인덱스를 비교하는 방법으로 정답을 구하였다.
import java.util.*;
class Solution {
public int[] solution(int n, String[] words) {
int[] answer = new int[2];
List<String> list = new ArrayList<>();
list.add(words[0]);
char target = words[0].charAt(words[0].length() - 1);
for (int i = 1; i < words.length; i++) {
if (!list.contains(words[i]) && target == words[i].charAt(0)) {
list.add(words[i]);
target = words[i].charAt(words[i].length() - 1);
} else {
answer[0] = i % n + 1;
answer[1] = i / n + 1;
break;
}
}
return answer;
}
}
중요한 부분은 if
문과 else
문인데, 우선 if
문에서는 끝말잇기 조건을 만족하는지에 대한 조건과 중복되는 단어가 list
에 담겨있는가를 체크하였다.
그 후 else
문에서는 몇 번째 사람이 몇 번째 차례에서 오류를 범했는지 구하는 식인데, list
에 words[0]
을 담고 시작했기에 i
가 1부터 시작하는 것을 감안하여 위와 같은 식으로 구성하였다.