프로그래머스 숫자야구 Swift

문제 설명

숫자 야구 게임이란 2명이 서로가 생각한 숫자를 맞추는 게임입니다.

각자 서로 다른 1~9까지 3자리 임의의 숫자를 정한 뒤 서로에게 3자리의 숫자를 불러서 결과를 확인합니다. 그리고 그 결과를 토대로 상대가 정한 숫자를 예상한 뒤 맞힙니다.

* 숫자는 맞지만, 위치가 틀렸을 때는 볼 * 숫자와 위치가 모두 맞을 때는 스트라이크 * 숫자와 위치가 모두 틀렸을 때는 아웃

예를 들어, 아래의 경우가 있으면

A : 123 B : 1스트라이크 1볼. A : 356 B : 1스트라이크 0볼. A : 327 B : 2스트라이크 0볼. A : 489 B : 0스트라이크 1볼.

이때 가능한 답은 324와 328 두 가지입니다.

질문한 세 자리의 수, 스트라이크의 수, 볼의 수를 담은 2차원 배열 baseball이 매개변수로 주어질 때, 가능한 답의 개수를 return 하도록 solution 함수를 작성해주세요.

제한사항

• 질문의 수는 1 이상 100 이하의 자연수입니다.
• baseball의 각 행은 [세 자리의 수, 스트라이크의 수, 볼의 수] 를 담고 있습니다.

입출력 예

baseball	return
[[123, 1, 1], [356, 1, 0], [327, 2, 0], [489, 0, 1]]	2

풀이: 이번 문제는 완전탐색입니다.

3자리 수이고 중복이 없으며(각자 서로 다른) 0(1~9)이 포함되지 않은 모든 숫자를 탐색합니다.

먼저 answer을 baseball안의 배열의 숫자만큼 이중배열으로 만들어줍니다. ->  var answer = Array(repeating: [Int](), count: baseball.count)

123~987까지의 숫자를 map으로 나눠줍니다. ->  let number =  String(i).map{String($0)}

이 숫자 중 0이 포함되어있다면 -> number.contains("0")

중복이 없다면 -> number[0] == number[1] || number[0] == number[2] || number[1] == number[2]

baseball안의 숫자만큼 for문을 돌려줍니다. -> for j in 0..<baseball.count{...}

strike와 ball의 수를 담을 변수를 만들어줍니다. ->   var strike = 0,ball = 0

baseball안에 0번째 있는 숫자를 map으로 나눠줍니다. -> let baseballmap = String(baseball[j][0]).map{String($0)}

baseball의 0번째 숫자가 3자리 수이기 때문에 for문으로 0~2까지 돌려줍니다. ->  for n in 0...2{...}

number의 위치와 baseball의 0번째 숫자 위치가 같다면 strike를 1 증가시켜줍니다. ->  if number[n] == baseballmap[n]{ strike += 1 }

위치가 같지 않지만 그 baseball의 0번째 숫자를 맵으로 나눈 숫자 중 number안의 숫자가 포함되어 있다면 ball을 1 증가시켜줍니다. -> else{ if baseballmap.contains(number[n]){ ball += 1 }}

그리곤 baseball의 배열 중 1번째 숫자와 strike가 같고 baseball의 2번째 숫자가 ball과 같다면 if baseball[j][1] == strike && baseball[j][2] == ball{...}

마지막으로 각 배열에 담겨있는 가능한 숫자들 중 공통된 숫자만 모으기 위해 ->   if j == 0 { answer[j].append(i) }else { if answer[j-1].contains(i){ answer[j].append(i) }} 를 해준 뒤 answer의 맨 마지막배열의 숫자를 return해줍니다.

 

전체코드

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34

import Foundation   func solution(_ baseball:[[Int]]) -> Int {     var answer = Array(repeating: [Int](), count: baseball.count)     for i in 123...987{         let number =  String(i).map{String($0)}         if number[0] == number[1] || number[0] == number[2] || number[1] == number[2] || number.contains("0"){             continue         }         for j in 0..<baseball.count{              var strike = 0,ball = 0             let baseballmap = String(baseball[j][0]).map{String($0)}             for n in 0...2{                 if number[n] == baseballmap[n]{                     strike += 1                 }else{                     if baseballmap.contains(number[n]){                     ball += 1                     }                 }             }             if baseball[j][1] == strike && baseball[j][2] == ball{                 if j == 0 {                 answer[j].append(i)                 }else{                     if answer[j-1].contains(i){                         answer[j].append(i)                     }                 }             }         }     }     return answer[answer.count-1].count } Colored by Color Scripter

cs