프로그래머스 2. 추억 점수

 사진들을 보며 추억에 젖어 있던 루는 사진별로 추억 점수를 매길려고 합니다. 사진 속에 나오는 인물의 그리움 점수를 모두 합산한 값이 해당 사진의 추억 점수가 됩니다. 예를 들어 사진 속 인물의 이름이 [“may”, “kein”, “kain”]이고 각 인물의 그리움 점수가 [5점, 10점, 1점]일 때 해당 사진의 추억 점수는 16(5 + 10 + 1)점이 됩니다. 다른 사진 속 인물의 이름이 [“kali”, “mari”, “don”, “tony”]이고 [“kali”, “mari”, “don”]의 그리움 점수가 각각 [11점, 1점, 55점]]이고, “tony”는 그리움 점수가 없을 때, 이 사진의 추억 점수는 3명의 그리움 점수를 합한 67(11 + 1 + 55)점입니다. 그리워하는 사람의 이름을 담은 문자열 배열 name, 각 사람별 그리움 점수를 담은 정수 배열 yearning, 각 사진에 찍힌 인물의 이름을 담은 이차원 문자열 배열 photo가 매개변수로 주어질 때, 사진들의 추억 점수를 photo에 주어진 순서대로 배열에 담아 return하는 solution 함수를 완성해주세요.

1차 시도

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var yearningNames = [String : Int]()
    var result = [Int]()
		
    for i in 0...name.count - 1 {
        yearningNames[name[i]] = yearning[i]
    }
		
    for j in 0...photo.count - 1 {
        var yearningPoint = Int()
        for k in 0...photo[j].count - 1 {
            if yearningNames[photo[j][k]] == nil {
                yearningPoint += 0
            } else {
                yearningPoint += yearningNames[photo[j][k]]!
            }
        }
        result.append(yearningPoint)
    }
    return result
}

결과

테스트 2개 시간 초과

그리움 점수가 없는 name을 탐색할 때마다 yearningNames 딕셔너리에 0으로 업데이트해서 탐색 연산을 줄여보자.

2차 시도

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var yearningNames = [String : Int]()
    var result = [Int]()
		
    for i in 0...name.count - 1 {
        yearningNames[name[i]] = yearning[i]
    }
		
    for j in 0...photo.count - 1 {
        var yearningPoint = Int()
        for k in 0...photo[j].count - 1 {
            if yearningNames[photo[j][k]] == nil {
                yearningNames[photo[j][k]] = 0
                yearningPoint += 0
            } else {
                yearningPoint += yearningNames[photo[j][k]]!
            }
        }
        result.append(yearningPoint)
    }
    return result
}

결과

테스트 1개 시간 초과

희한하게도 1차 시도때 초과한 두 개 테스트는 통과하고 통과했던 테스트 중 하나에서 시간초과가 뜬다.

3차 시도

func solution(_ name:[String], _ yearning:[Int], _ photo:[[String]]) -> [Int] {
    var yearningNames = [String : Int]()
    var result = [Int]()

    for i in 0...name.count - 1 {
        yearningNames[name[i]] = yearning[i]
    }
	    
    for j in 0...photo.count - 1 {
        var yearningPoint = Int()
        for k in 0...photo[j].count - 1 {
            if yearningNames[photo[j][k]] == nil {
                yearningNames[photo[j][k]] = 0
            } else {
                yearningPoint += yearningNames[photo[j][k]]!
            }
        }
        result.append(yearningPoint)
    }
    return result
}

결과

yearningNames에 0을 더한다는 쓸데없는 연산을 줄여서, 테스트에 통과했다.