[프로그래머스]level.2 - 할인 행사(C++)

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

문제 설명

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

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

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

내 코드

첫번째 풀이

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

using namespace std;

int solution(vector<string> want, vector<int> number, vector<string> discount) {
    int answer = 0;
    map<string, int> shopping_list;
    
    // 쇼핑 리스트 채우기
    for(int i = 0; i < want.size(); ++i)
    {
        shopping_list.insert({want[i], number[i]});
    }
    
    // 모두 할인 받을 수 있는 날의 수 찾기
    for(int i = 0; i < discount.size(); ++i)
    {
        map<string, int> copy = shopping_list;
        
        int last_day = i+10;
        if(last_day > discount.size())
            last_day = discount.size();
        
        for(int d = i; d < last_day;  ++d)
        {        
            auto iter = copy.find(discount[d]);
            if(iter != copy.end())
            {   
                --copy[iter->first];
                if(copy[iter->first] <= 0)
                    copy.erase(iter);
            }
        }
        
        if(copy.empty())
            ++answer;    
    }
    
    return answer;
}

두번째 풀이

첫번째 풀이를 바탕으로 조금 더 조건을 추가해서 반복을 최소화하여 시간을 조금 더 줄인 코드

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

using namespace std;

int solution(vector<string> want, vector<int> number, vector<string> discount) {
    int answer = 0;
    map<string, int> shopping_list;
    int total = 0;
    
    // 쇼핑 리스트(map) 채우기
    for(int i = 0; i < want.size(); ++i)
    {
        shopping_list.insert({want[i], number[i]});
        total += number[i];
    }
    
    // 각 할인 행사 날을 시작일로 설정하여 10일 동안 물건을 모두 살 수 있는지 검사한다.
    for(int i = 0; i < discount.size(); ++i)
    {
        // 쇼핑 리스트 복제
        map<string, int> copy = shopping_list;
        
        // 마지막 날 계산
        int last_day = i+10;
        if(last_day > discount.size())
            last_day = discount.size();
        
        // 남은 날이 쇼핑 리스트의 물건의 수보다 적을 경우, 모든 물건을 사지 못하므로
        // 이후의 날들에 대해 더 이상 검사를 진행하지 않고 넘어간다.
        if(last_day-i < total)
            break;
        
        // 10일간 할인 품목 검사
        for(int d = i; d < last_day;  ++d)
        {
            // 사고자 하는 물건이 쇼핑 리스트에 있다면 map에서 해당 물건의 개수를 줄인다. (0이 되면 뺀다)
            auto iter = copy.find(discount[d]);
            if(iter != copy.end())
            {   
                --copy[iter->first];
                if(copy[iter->first] <= 0)
                    copy.erase(iter);
            }
            
            // 쇼핑 리스트에 더 이상 물건이 남지 않았다면 검사를 종료한다.
            if(copy.empty())
                break;
        }
        
        // 쇼핑 리스트가 비었다면 모든 물건을 살 수 있으니 정답으로 추가한다.
        if(copy.empty())
            ++answer;    
    }
    
    return answer;
}

테케 10번과 8번을 비교하면 실행 시간이 준 것을 확인할 수 있다.