[Swift] 추억 점수

[문제]

코딩테스트 연습 - 추억 점수 | 프로그래머스 스쿨 (programmers.co.kr)

프로그래머스

[문제 풀이]

딕셔너리를 이용해서 쉽게 풀 수 있는 문제이다.

딕셔너리를 만들어 name에 yearning을 매칭시켜주고

photo를 방문하면서 총 숫자가 몇이 나오는지를 answer에 저장해주자.

[회고]

.

[코드]

import Foundation

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var answer = [Int]()
    var dict = [String: Int]()
    //name과 yearning을 돌면서 딕셔너리에 넣자.
    
    for i in 0..<name.count {
        dict[name[i]] = yearning[i]
    }
    
    for i in photo {
        var temp = 0
        for j in i {
            temp += dict[j] ?? 0
        }
        answer.append(temp)
    }
    
    return answer
}