1부터 n까지 번호가 붙어있는 n명의 사람이 영어 끝말잇기를 하고 있습니다. 영어 끝말잇기는 다음과 같은 규칙으로 진행됩니다.
tank → kick → know → wheel → land → dream → mother → robot → tank
위 끝말잇기는 다음과 같이 진행됩니다.
사람의 수 n과 사람들이 순서대로 말한 단어 words 가 매개변수로 주어질 때, 가장 먼저 탈락하는 사람의 번호와 그 사람이 자신의 몇 번째 차례에 탈락하는지를 구해서 return 하도록 solution 함수를 완성해주세요.
import java.util.ArrayList;
class Solution {
public int[] solution(int n, String[] words) {
int count = 1;
int[] answer = {};
ArrayList<String> list = new ArrayList<>(); //list 선언
list.add(words[0]);
for (int i = 1; i < words.length; i++) {
if (words[i].split("")[0].equals(words[i - 1].split("")[words[i - 1].length() - 1])) {
// 끝말잇기가 되는지 안되는지 확인
for (int j = 0; j < list.size(); j++) {
if (words[i].equals(list.get(j))) {
count--;
break;
// 같은단어를 했는지 안했는지
}
}
list.add(words[i]);
count++;
} else {
break; //끝말잇기 실패시
}
}
if (count == words.length) answer = new int[]{0, 0}; //끝말잇기 성공했을경우
else answer = new int[]{count % n + 1, count / n + 1}; // 끝말잇기 실패했을경우
return answer;
}
}
import java.util.*;
class Solution {
public int[] solution(int n, String[] words) {
int[] answer = {};
int cnt = 1;
Set<String> set = new HashSet<>();
set.add(words[0]);
for(int i=1; i<words.length; i++) {
if(set.contains(words[i])) break;
if(words[i].charAt(0)==words[i-1].charAt(words[i-1].length()-1)) {
set.add(words[i]);
cnt++;
}
else break;
}
if(cnt==words.length) answer = new int[] {0,0};
else answer = new int[] {cnt%n+1, cnt/n+1};
return answer;
}
}