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

 

🍄 프로그래머스 문제 링크

프로그래머스

✍🏻 문제 풀이 Github 링크

-Swift-CodingTest/프로그래머스/2/131127. 할인 행사 at main · BaeJihae/-Swift-CodingTest

 

 

문제 설명

 

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

 

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

 

정현이가 원하는 제품을 나타내는 문자열 배열 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

 

 

문제 풀이

 

나의 풀이

import Foundation

func solution(_ want:[String], _ number:[Int], _ discount:[String]) -> Int {
    
    var wantDic = [String:Int]()
    var count = 0
    
    for (product, num) in zip(want, number) {
        wantDic[product] = num
    }
    
    for i in 0..<discount.count-9 {
        var dic = [String:Int]()
        for j in 0...9 {
            dic[discount[i+j]] = ( dic[discount[i+j]] ?? 0 ) + 1
        }
        if dic == wantDic {
            count += 1
        }
    }
    
    return count
}

 

discount의 배열을 돌면서 [0…9] 까지의 배열들을 딕셔너리로 바꾸어 정현이가 원하는 제품과 일치하는지 파악해야했다. 그 과정에서 딕셔너리를 사용하여 두 개의 딕셔너리가 같다면 count를 올리도록 로직을 구현했다.

 

discount의 최대 길이가 100,000이므로 최대 100,000-10의 for문을 돌며 배열을 딕셔너리에 정리하는 과정을 거쳤다. 하나의 배열을 딕셔너리에 넣는 경우의 수는 10 이므로 최대 ( 100,000-10 ) * 10 번을 계산해야한다.

 

또한 정현이가 원하는 제품과 원하는 제품의 갯수를 딕셔너리에 담는 과정은 최대 10번이다.

그러니 나의 풀이의 최대 반복 횟수는 10 + ( 100,000-10 ) * 10 이라고 판단하여 충분히 풀리겠다고 판단했다.

 

 

다른 사람 풀이

import Foundation

func solution(_ want:[String], _ number:[Int], _ discount:[String]) -> Int {
    let cnt = want.count
    var wantArr = [Int](repeating: 0, count: cnt)
    var discountArr = [Int](repeating: 0, count: cnt)

    var saleDict = [String: Int]()

    for i in 0..<want.count {
        saleDict[want[i]] = i
        wantArr[i] = number[i]
    }

    for i in 0..<10 {
        guard let idx = saleDict[discount[i]] else {
            continue
        }
        discountArr[idx] += 1
    }

    var answer = 0

    if wantArr == discountArr {
        answer += 1
    }

    for i in 0..<discount.count-10 {
        if let idx = saleDict[discount[i]] {
            discountArr[idx] -= 1
        }

        if let idx = saleDict[discount[i+10]] {
            discountArr[idx] += 1
        }

        if wantArr == discountArr {
            answer += 1
        }
    }

    return answer
}

 

이 분의 풀이는 처음 0~9까지의 배열의 개수를 계산해서 Arr에 넣은 후에 그 다음 1~10까지의 배열을 비교하기 위해서는 원래의 0의 개수를 빼고, 그다음에 올 10의 개수를 더하는 식으로 로직을 구현했다. 그러니 나의 계산보다 훨~씬 빨랐다. 나는 0~9까지의 배열을 반복적으로 딕셔너리에 넣는 다면 이 분의 풀이는 바뀐 배열만 계산하여 시간 복잡도도 공간 복잡도도 훨씬 빨랐다.

 

처음엔 풀이가 이해가 되지 않아 하나하나 뜯어본 뒤 스스로 다시 이 풀이로 풀어보았다. wandDic은 굳이 만들지 않아도 count랑 비교하여 계산하면 되었고, 특이하게 딕셔너리에는 정현이가 원하는 제품들의 index값을 넣어서 discount를 비교할때 해당 제품이 딕셔너리에 들어있으면 해당 딕셔너리의 value값 ( 정현이가 원하는 제품의 index 값 ) 에 있는 배열의 값을 올렸다.

 

다른 사람 풀이를 참고하여 다시 풀어본 나의 풀이

func solution(_ want:[String], _ number:[Int], _ discount:[String]) -> Int {
    
    var wantArrNum = [String: Int]()
    var discountArr = [Int](repeating: 0, count: want.count)
    var count = 0
    
    for i in 0..<want.count {
        wantArrNum[want[i]] = i
    }
    
    for i in 0..<10 {
        guard let tmp = wantArrNum[discount[i]] else {
            continue
        }
        discountArr[tmp] += 1
    }
    
    print(number, discountArr)
    
    if number == discountArr {
        count += 1
    }
    
    for i in 0..<discount.count-10 {
        if let arrNum = wantArrNum[discount[i]] {
            discountArr[arrNum] -= 1
        }
        
        if let arrNum = wantArrNum[discount[i+10]]  {
            discountArr[arrNum] += 1
        }
        
        print(number, discountArr)
        
        if number == discountArr {
            count += 1
        }
    }
    
    return count
}