[Swift] 로또의 최고 순위와 최저 순위

[문제]

코딩테스트 연습 - 로또의 최고 순위와 최저 순위 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

[문제 풀이]

딕셔너리를 이용해서 로또 맞춘 숫자와 순위를 지정해두고

lottos 안에 0이나 해당 숫자가 있으면 최고 숫자를 올리고

lottos 안에 해당 숫자가 있을때만 낮은 숫자를 올린다.

마지막에 answer 안에 딕셔너리를 이용해 최고 숫자의 순위, 최저 숫자의 순위를 넣어주자.

[회고]

var arr = [1,2,3,4,5,6]
arr.contains(1) = true

 

contains 함수를 이용하면 쉽게 풀린다.

contains 함수는 해당 배열 안에 해당 변수가 있는지를 true / false로 알려주는 함수이다.

[코드]

import Foundation

func solution(_ lottos:[Int], _ win_nums:[Int]) -> [Int] {
    var answer = [Int]()
    var high = 0
    var bottom = 0
    
    var dict = [Int: Int]()
    dict[6] = 1
    dict[5] = 2
    dict[4] = 3
    dict[3] = 4
    dict[2] = 5
    dict[1] = 6
    dict[0] = 6
    
    for i in lottos {
        if i == 0 {
            high += 1
        }
        else {
            //만일 갖고 있다면
            if win_nums.contains(i) == true {
                high += 1
                bottom += 1
            }
        }
    }
    
    answer.append(dict[high]!)
    answer.append(dict[bottom]!)
    
    return answer
}