10. 프로그래머스 - 할인 행사(해시)

문제 설명
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: ["banana", "apple", "rice", "pork", "pot"]
number: [3, 2, 2, 2, 1]
discount: ["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"]
result: 3

want: ["apple"]
number: [10]
discount: ["banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana", "banana"]
result: 0

 

입출력 예 설명

입출력 예 #1
문제 예시와 같습니다.

입출력 예 #2
사과가 할인하는 날이 없으므로 0을 return 합니다.

---

특정일에 회원 가입을 해 그날부터 10일간 쇼핑할 때 구매하려는 제품을 모두 할인받아 살 수 있는지 확인하고 특정일을 세어 반환하면 된다. 

 

입력값 분석하기

want는 구매하고자 하는 제품, number는 want를 얼마나 사고 싶은지(개수)를 나타낸다. discount에는 마트에서 각 날짜마다 할인하는 제품 정보가 들어 있다. 

입출력 예를 보면 want와 numbers는 연관이 있는 것처럼 보인다.

• want[0] = "banana"
• number[0] = 3

이를 연결 지어 생각하면 HashMap.put(want[0], number[0]), want[0] : "banana", number[0] : 3 즉, 바나나를 3개 사고 싶다는 걸 알 수 있다.

 

그 다음엔 특정일에 회원가입을 하면 할인 받을 수 있는 품목과 품목의 개수를 해시맵으로 만들면 10일동안 할인 받을 수 있는 품목과 그 품목을 할인 받을 수 있는 개수를 알 수 있다.

 

이로써 우리가 해시맵으로 표현한 것은 다음과 같다.

• 내가 사려고 하는 품목과 품목의 개수
• n일 차에 회원가입을 하면 할인받아 살 수 있는 품목과 품목의 개수

두 항목은 해시맵이므로 equals() 메서드로 둘이 일치하는지 확인할 수 있다.

 

알고리즘 코드

import java.util.HashMap;

class Solution {
    public int solution(String[] want, int[] number, String[] discount) {
        HashMap<String, Integer> wantMap = new HashMap<>();
        
        for(int i = 0; i < want.length; i++) {
            wantMap.put(want[i], number[i]);
        }
    
        int answer = 0; 
        
        // 특정일 i에 회원가입 시 할인 받을 수 있는 품목 체크
        for(int i = 0; i < discount.length - 9; i++) {
        	// i일 회원가입 시 할인 받는 제품 및 개수를 담을 해시맵
            HashMap<String, Integer> discountItemMap = new HashMap<>();
            
            // i일 회원가입 시 할인받는 제품 및 개수로 해시맵 구성
            for(int j = i; j < i + 10; j++) {
            	// 원하는 품목에 dicount 제품이 없을 땐 건너뛰면 된다.
                if(wantMap.containsKey(discount[j])) {
                    discountItemMap.put(discount[j], discountItemMap.getOrDefault(discount[j], 0) + 1);
                }
            }
            
            if(discountItemMap.equals(wantMap))
                answer++;
        }
        return answer;
    }
}

 

시간 복잡도 분석

N은 discount 배열의 길이고 주어진 want 배열에 기반해 10일 동안 할인 상품이 원하는 제품과 이치하는지 확인 하므로 시간 복잡도는 O(N)이다.