66-9kyo-hwang

[9kyo-hwang]

🔗 문제 링크

할인 행사

✔️ 소요된 시간

30분

✨ 수도 코드

1. 문제

10일 동안 회원 자격이 부여된다. 회원은 매일 한 가지 제품을 할인받을 수 있다. 제품은 하루에 하나만 구매할 수 있다.
예를 들어, 14일간 [치킨, 사과, 사과, 바나나, 쌀, 사과, 돼지고기, 바나나, 돼지고기, 쌀, 냄비, 바나나, 사과, 바나나] 순서대로 제품을 할인한다고 하자. 나는 바나나 3개, 사과 2개, 쌀 2개, 돼지고기 2개, 냄비 1개를 할인해서 사고 싶다.

1. 첫째 날부터 열흘 간에는 냄비가 할인하지 않기 때문에 첫째 날에는 회원가입을 하지 않는다.
2. 둘째 날부터 열흘 간에는 바나나를 원하는 만큼 할인구매할 수 없기 때문에 둘째 날에도 회원가입을 하지 않는다.
3. 셋째 날, 넷째 날, 다섯째 날부터 각각 열흘은 원하는 제품과 수량이 일치한다.

원하는 제품을 모두 할인 받을 수 있는 날짜는 셋째 날, 넷째 날, 다섯째 날로 총 3일이다.
이와 같이 회원등록시 원하는 제품을 모두 할인 받을 수 있는 회원등록 날짜의 총 일수를 return 하는 solution 함수를 완성해야 한다. 가능한 날이 없으면 0을 return 한다.

2. 설명

주어지는 총 제품 개수가 최대 100,000개이기 때문에, 매번 10일 간 할인하는 품목 수량을 계산하기엔 시간이 너무 오래 걸린다.
할인 기간이 "10일"로 고정되어 있기 때문에, 여기서는 슬라이딩 윈도우 기법을 사용하는 것이 적절하다.

단, 여기서는 문자열 원소를 카운팅해야 하기 때문에 map 자료구조를 사용해 응용한다.

int solution(vector<string> InWants, vector<int> InNumbers, vector<string> InDiscounts) 
{
    unordered_map<string, int> NumberbyWants;
    for(int i = 0; i < 9; ++i)
    {
        NumberbyWants[InDiscounts[i]]++;  // 9일까지의 할인되는 제품들의 개수를 카운팅한다.
    }

    ...
}

string, 즉 제품 문자열을 key로 해서 각각의 제품들이 몇 개 할인 중인지 기록하는 NumberbyWants를 선언한다.
회원 자격이 열흘 유지되기 때문에, 우선 최초의 9일 할인 제품들을 기록한다.

    int Answer = 0;
    for(int i = 9; i < InDiscounts.size(); ++i)
    {
        NumberbyWants[InDiscounts[i]]++;  // 현재 할인 품목 map에 기록(right pointer++)
        bool Flag = true;
        
        for(int j = 0; j < InWants.size(); ++j)  // Wants 배열을 순회하며
        {
            if(NumberbyWants[InWants[j]] != InNumbers[j])  // Wants 항목의 수량(Number)와 일치하지 않다면 이 기간은 조건 불충족
            {
                Flag = false;
                break;
            }
        }
        
        Answer += Flag;
        NumberbyWants[InDiscounts[i - 9]]--;  // 첫째 날 할인 품목 제외(left pointer++)
    }

    return Answer;

이제 10일째부터 슬라이딩 윈도우 기법을 적용한다.

• 현재 날짜의 할인 제품을 map에 카운팅한다.
• map을 순회하며 내가 원하는 품목의 개수와 하나라도 일치하지 않는다면 Flag를 false로 하고 순회를 중단한다.
• 만약 반복문이 종료됐는데도 Flag가 true라면 지금까지의 열흘은 나의 조건을 충족하는 것이므로 Answer += 1을 한다.
• 마지막으로 첫째 날 할인하는 품목을 map에서 제거한다.

전체 코드

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

int solution(vector<string> InWants, vector<int> InNumbers, vector<string> InDiscounts) 
{
    unordered_map<string, int> NumberbyWants;
    for(int i = 0; i < 9; ++i)
    {
        NumberbyWants[InDiscounts[i]]++;
    }
    
    int Answer = 0;
    for(int i = 9; i < InDiscounts.size(); ++i)
    {
        NumberbyWants[InDiscounts[i]]++;
        bool Flag = true;
        
        for(int j = 0; j < InWants.size(); ++j)
        {
            if(NumberbyWants[InWants[j]] != InNumbers[j])
            {
                Flag = false;
                break;
            }
        }
        
        Answer += Flag;
        NumberbyWants[InDiscounts[i - 9]]--;
    }
    
    return Answer;
}

📚 새롭게 알게된 내용

사실 처음 풀었을 땐 누적합 방식을 써서 개수를 구했는데, 생각해보니 슬라이딩 알고리즘으로 개수 세는 걸 실시간으로 빠르게 할 수 있을 것 같아 코드를 수정했다.

[mjj111]

딕셔너리랑 슬라이딩윈도우 같이 푸는 문제라 재밌네요!!!!

import collections

def can_membership(foods, want):
    for food in want:
        if foods[food] > 0:  
            return False
    return True

def solution(want, number, discount):
    foods = collections.defaultdict(int)
    
    for food, count in zip(want, number):
        foods[food] = count

    n = len(discount)
    
    for food in discount[:10]:
        foods[food] -= 1

    answer = 0
    start = 0
    end = 9
    
    while True:
        if can_membership(foods, want):
            answer += 1
        
        if end + 1 == n:  
            break
        foods[discount[start]] += 1 
        
        start += 1
        end += 1
        foods[discount[end]] -= 1 
        
    return answer