프로그래머스 - 위장(Swift)

프로그래머스 - 위장(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/42578

프로그래머스

 

 나의 풀이

 

다른 옷을 조합해서 입는 경우의 수는 카테고리마다 이 카테고리에 해당하는 의상을 입지 않는다의 경우와 이 카테고리에 몇개의 의상이 있는지만 고려하면 된다.

 

예를들어 상의에는 2벌, 하의에는 3벌이 있을때 상의를 아예 안 입거나 1번 혹은 2번을 고르는 경우 -> 3, 하의를 아예 안 입거나 1번 혹은 2번 혹은 3번을 고르는 경우 -> 4해서 총 12가지 경우의 수가 나온다. 여기서 한 가지는 꼭 입어야하는 조건이 있으니 -1만 해주면 된다.

 

import Foundation

func solution(_ clothes:[[String]]) -> Int {
    var dic: [String: Int] = [:]
    
    for cloth in clothes {
        if dic[cloth[1]] == nil {
            dic[cloth[1]] = 1
        } else {
            dic[cloth[1]]! += 1
        }
    }
    var result = 1
    for x in dic {
        result *= (x.1+1) 
    }
    
    return result-1
}

 

 다른 사람의 풀이

 

compactMap, Set, reduce를 적절히 사용한 풀이이다.

 

compactMap은 보통은 2차원배열에서 사용하면 2차원 배열로 리턴을 하지만 아래 코드처럼 $0.last!로 한 개의 숫자만 리턴하게 된다면 1차원 배열로 리턴하게 된다. (왜인지는 모르겠습니다. 혹시 알고 계시면 댓글 부탁드려요!) 

 

그리고 고차함수를 쓸때 어느 타이밍에 filter를 쓰고 reduce를 써야할지 헷갈릴 수 있는데 항상 "내가 무엇을 만드는가"에 집중하면 된다.

 

아래 예시에서는 각각 의상의 종류에 따라 카운트를 누적시키고 싶었기 때문에 catories를 매핑했고 clothes를 전부 돌면서 해당하는 것을 filter -> count했다. 

 

import Foundation

func solution(_ clothes:[[String]]) -> Int {
    let types = clothes.compactMap{ $0.last! }
    let setType = Set(types)
    let categories = Array(setType)
    
    let counts = categories.map { category in
        return clothes.filter{ $0.last! == category }.count + 1
    }
    
    return counts.reduce(1, {$0 * $1}) - 1  
}

 

 피드백

 

딕셔너리를 for문으로 접근할 때 튜플처럼 .연산자로 key나 value에 접근이 가능하다.

 

let dic: [String: Int] = ["a": 1, "b": 2]
for x in dic {
	print(x.1 terminator:" ") // output: 1 2
}