문제 131127. 할인 행사

문제 링크

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

프로그래머스

 

난이도 : Lv.2

 

문제 요약 설명

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

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

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

입력

• 원하는 제품을 나타내는 문자열 배열 want
• 정수 배열 number
• XYZ 마트에서 할인하는 제품을 나타내는 문자열 배열 discount

입력 제한

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

입력 예제

// input
["banana", "apple", "rice", "pork", "pot"] // want
[3, 2, 2, 2, 1] // number
["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"] // discount

// ans
result // 3

 

풀이 방식

정확히 10일 내에 원하는 제품들을 수량만큼 전부 살 수 있는 지 체크해야한다. 또한, 구매할 물건의 수량의 합은 무조건 10이다. 즉, 크기가 10인 연속된 수열에서 모든 물건을 수량에 맞게 사면 된다. 

naive하게 생각하면 이 문제의 시간복잡도는 O(n^2)이 된다. 시작일을 첫 날로 잡고 10일간의 물품들을 전부 수량에 맞게 살 수 있는 지 확인하고, 그렇다면 정답으로 리턴, 그렇지 않으면 둘째날을 시작일로 잡고 10일간의 물품들을 전부 수량에 맞게 살 수 있는 지 확인한다. 이 과정은 discount()-10인날까지 반복한다. 그러나 이 방식은 날에 할인하는 물품을 최대 10면 중복하여 구하게 된다.(비효율적)

 

그래서 다음 날로 넘어갈 때, 시작일의 할인 정보를 빼고 나머지 할인정보를 다음 시작일에 사용하면 정답을 효율적으로 구할 수 있다. 이를 위해 map자료구조(변수 m)를 이용해 10일간의 물품과 해당 물품의 수량을 저장하고, 다음 시작일로 넘어갈 때마다 시작일의 할인정보를 빼고 다음 시작일+9인 날의 할인정보를 추가한다.

 

이제 m을 이용해 10일 동안 구매한 물품들의 수량을 전부 구매할 수 있는 지 확인하여 정답을 구하면 된다.

 

정답 코드 

#include <string>
#include <vector>
#include <map>
#include <iostream>

using namespace std;

int solution(vector<string> want, vector<int> number, vector<string> discount) {
    int answer = 0;
    map<string, int> m;
    
    for(int i = 0; i < discount.size(); i++){
        if(i < 9){
            m[discount[i]]++;
            continue;
        }
        
        m[discount[i]]++;
        bool find = true;
        for(int n = 0; n < want.size(); n++){
            if(m[want[n]] != number[n]){
                find = false;
                break;
            } 
        }
        
        if(find){
            answer++;
            cout << i << '\n';
        } 
        m[discount[i-9]]--;
    }
    return answer;
}