프로그래머스 - 할인 행사[Swift]

할인 행사

문제 설명

XYZ 마트는 일정한 금액을 지불하면 10일 동안 회원 자격을 부여합니다. XYZ 마트에서는 회원을 대상으로 매일 한 가지 제품을 할인하는 행사를 합니다. 할인하는 제품은 하루에 하나씩만 구매할 수 있습니다. 알뜰한 정현이는 자신이 원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치할 경우에 맞춰서 회원가입을 하려 합니다.

예를 들어, 정현이가 원하는 제품이 바나나 3개, 사과 2개, 쌀 2개, 돼지고기 2개, 냄비 1개이며, XYZ 마트에서 15일간 회원을 대상으로 할인하는 제품이 날짜 순서대로 치킨, 사과, 사과, 바나나, 쌀, 사과, 돼지고기, 바나나, 돼지고기, 쌀, 냄비, 바나나, 사과, 바나나인 경우에 대해 알아봅시다. 첫째 날부터 열흘 간에는 냄비가 할인하지 않기 때문에 첫째 날에는 회원가입을 하지 않습니다. 둘째 날부터 열흘 간에는 바나나를 원하는 만큼 할인구매할 수 없기 때문에 둘째 날에도 회원가입을 하지 않습니다. 셋째 날, 넷째 날, 다섯째 날부터 각각 열흘은 원하는 제품과 수량이 일치하기 때문에 셋 중 하루에 회원가입을 하려 합니다.

정현이가 원하는 제품을 나타내는 문자열 배열 want와 정현이가 원하는 제품의 수량을 나타내는 정수 배열 number, XYZ 마트에서 할인하는 제품을 나타내는 문자열 배열 discount가 주어졌을 때, 회원등록시 정현이가 원하는 제품을 모두 할인 받을 수 있는 회원등록 날짜의 총 일수를 return 하는 solution 함수를 완성하시오. 가능한 날이 없으면 0을 return 합니다.

---

제한사항

• 1 ≤ want의 길이 = number의 길이 ≤ 10
  • 1 ≤ number의 원소 ≤ 10
  • number[i]는 want[i]의 수량을 의미하며, number의 원소의 합은 10입니다.
• 10 ≤ discount의 길이 ≤ 100,000
• want와 discount의 원소들은 알파벳 소문자로 이루어진 문자열입니다.
  • 1 ≤ want의 원소의 길이, discount의 원소의 길이 ≤ 12

 

입출력 예

want                                         number         discount                                                                                                            result

["banana", "apple", "rice", "pork", "pot"]	[3, 2, 2, 2, 1]	["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"]	3
["apple"]	[10]	["banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana"]	0

 

---

풀이 및 소스코드

원하는 품목과 개수를 dictionary형태로 저장한다.

discount의 품목또한 dictionary형태로 저장하는데 10개까지만 저장한다.

joinable이란 함수는 a딕셔너리의count가 bdictionary 이상이면 true, 하나라도 부족하면 false를 리턴한다.

하루가 지날 때 마다 일자가 바뀌면서 전날의 할인상품은 빠지고, 10일뒤의 할인상품이 추가된다

딕셔너리의 개수를 날이 바뀔 때 마다 추가, 더해주고 joinable 함수를 통해 조건을 충족하는지 검사한다.

import Foundation
//일정 금액을 지불하면 10일동안 회원자격 부여
//회원을 대상으로 매일 한 가지를 할인, 하루에 하나씩만 구매가능
//원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치하는 경우에만 회원가입

func solution(_ want:[String], _ number:[Int], _ discount:[String]) -> Int {
    var wantdic = [String: Int]()
    var discntdic = [String: Int]()
    
    //원하는 상품 별 개수를 count함
    for idx in 0..<want.count{
        let name = want[idx]
        let cnt = number[idx]
        wantdic.updateValue(cnt, forKey: name)
    }
    
    //10개까지만 추가한다. 첫 비교는 10일만 하기 때문
    for idx in 0...9{
        let dis = discount[idx]
        discntdic[dis] = (discntdic[dis] ?? 0) + 1
    }
    
    //a의 모든 원소가 b의 원소 이상이면 true 아니면 false 리턴
    func joinable(_ a: [String: Int], _ b: [String: Int]) -> Bool{
        for item in a{
            let name = item.key
            let cnt = item.value
            
            if (b[name] ?? 0) < cnt{
                return false
            }
        }
        return true
    }
    
    var count = 0
    
    //첫 날이 아니면 전날의 품목은 빠지고, 10일 뒤의 품목은 추가
    for day in 0..<discount.count - 9{
        if day != 0{
            var removeitem = discount[day-1]
            var additem = discount[day+9]
            
            discntdic[removeitem] = (discntdic[removeitem] ?? 0) - 1
            discntdic[additem] = (discntdic[additem] ?? 0) + 1
        }
        count = joinable(wantdic, discntdic) ? count + 1 : count
    }
    
    return count
}