코니는 영어 단어가 적힌 카드 뭉치 두 개를 선물로 받았습니다. 코니는 다음과 같은 규칙으로 카드에 적힌 단어들을 사용해 원하는 순서의 단어 배열을 만들 수 있는지 알고 싶습니다.
예를 들어 첫 번째 카드 뭉치에 순서대로 ["i", "drink", "water"], 두 번째 카드 뭉치에 순서대로 ["want", "to"]가 적혀있을 때 ["i", "want", "to", "drink", "water"] 순서의 단어 배열을 만들려고 한다면 첫 번째 카드 뭉치에서 "i"를 사용한 후 두 번째 카드 뭉치에서 "want"와 "to"를 사용하고 첫 번째 카드뭉치에 "drink"와 "water"를 차례대로 사용하면 원하는 순서의 단어 배열을 만들 수 있습니다.
문자열로 이루어진 배열 cards1
, cards2
와 원하는 단어 배열 goal
이 매개변수로 주어질 때, cards1
과 cards2
에 적힌 단어들로 goal
를 만들 있다면 "Yes"를, 만들 수 없다면 "No"를 return하는 solution 함수를 완성해주세요.
fun solution(cards1: Array<String>, cards2: Array<String>, goal: Array<String>): String {
return if(goal.any { it in cards1 || it in cards2}) "Yes" else "No"
}
any를 사용해 cards1이나 cards2안에 goal의 원소가 포함되어 있으면 Yes 아니면 No를 리턴하게 만들었다
그런데 문제가 있었는데 조건중하나인 기존에 주어진 카드 뭉치의 단어 순서는 바꿀 수 없습니다.
를 생각하지 못했다
fun solution(cards1: Array<String>, cards2: Array<String>, goal: Array<String>): String {
var a = 0
var b = 0
for (index in goal){
when{
a < cards1.size && index == cards1[a] -> a++
b < cards2.size && index == cards2[b] -> b++
else -> return "No"
}
}
return "Yes"
}
변수 a, b를 추가해서 goal안의 index와 cards의 순서에 따라 비교하고 카드 사이즈를 초과하게되면 더이상 a, b가 증가하지 않게해 ArrayIndexOutOfBoundsException
를 예방했다