본문 바로가기

코딩 테스트(Coding test)/Lv. 2

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

미숙한 블로그 주인이 코딩테스트 문제를 풀어가는 과정을 담은 글입니다. 이 풀이가 효율적인 풀이가 아닐 수 있으며, 부정확한 정보가 많이 있을 수 있습니다. 보완해야할 점이 있다면 댓글로 남겨주세요!

 

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

할인 행사

문제

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

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

정현이가 원하는 제품을 나타내는 문자열 배열 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 배열에서 직접 확인하는 방법도 있겠지만, 효율적이지 못하므로 맵을 이용해 바로 접근해주도록 한다.

 

이후에는 문제에서 요구하는 대로 코드를 작성해준다.

 

1. 10일 단위로 끊어서 다음을 확인한다.

1-1. 할인 상품이 정현이가 원하는 것이 아니거나, 이미 다 구매했다면 제품과 수량이 일치하지 않는다.( = 0을 반환한다. )

- 즉, 맵에서 해당 key의 value가 0이면 0을 반환한다는 소리이다.

1-2. 그렇지 않으면, 해당 상품을 하나 구매하고 수량을 하나 낮춰준다.

1-3. 10개의 상품을 모두 구매했다면 제품과 수량이 일치하므로 그 날은 회원가입을 한다.( = 1을 반환한다. )

2. 회원가입 가능한 날짜를 모두 더해준다.

 

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

using namespace std;

// 회원가입 하려는 날인지?
int check(map<string, int> wlist, vector<string> &discount, int idx) {
    for (int i = idx; i < (idx + 10); i++) {
        // 제품과 수량이 일치하지 않으면 회원가입을 하지 않음.
        if (!wlist[discount[i]]) {
            return 0;
        }
        wlist[discount[i]]--;
    }
    // 제품과 수량이 일치해 회원가입을 함.
    return 1;
}

int solution(vector<string> want, vector<int> number, vector<string> discount) {
    int answer = 0;
    map<string, int> wlist;
    
    for (int i = 0; i < number.size(); i++) {
        wlist[want[i]] = number[i];
    }
    
    int d_size = discount.size();
    d_size -= 9;
    // 열흘 단위로 나눠서 회원가입을 할지 판별
    for (int i = 0; i < d_size; i++) {
        answer += check(wlist, discount, i);
    }
    
    return answer;
}