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

문제

할인 행사 : 문제 링크

문제 분석

• 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 내에 discount의 원소가 존재하는지 확인하기 위해 find() 함수를 사용하므로 algorithm 헤더를 include
• 원하는 제품을 모두 할인 받을 수 있는 날짜의 총 일수를 저장할 정수형 변수 answer을 0으로, number의 원소를 복사하여 저장할 정수형 vector original_number을 초기화. 10일 연속으로 일치할 경우를 찾는 것이므로 외부 for loop의 조건문을 i <= discount.size() - 10으로 설정하고, 10일 모두 할인 받을 수 있는지 확인할 정수형 변수 check를 1로, number에 original_number을 복사하여 저장
• 내부 for loop의 초기화식을 j = i, 조건식을 j < i + 10으로 설정하여 10개의 원소만 순회. 코드 가독성을 위해 auto형 변수 found에 find() 함수를 사용하여 want내에 현재 인덱스의 discount 원소가 있는지 확인한 반환자를 저장. if문을 사용하여 found가 want.end()가 아니고 (want내에 discount 원소를 찾음) number의 동일한 인덱스의 원소가 0이 아니라면 그 원소를 1씩 낮춤. else if문을 사용하여 found가 want.end() 이거나 (want내에 discount 원소를 찾지 못함) number의 동일한 인덱스의 원소가 0이라면 (할인을 받을 수 없음) check를 0으로 저장하고 loop 탈출. 내부 loop 탈출 후, if문을 사용하여 check가 1이라면 10일동안 제품을 모두 할인 받을 수 있음을 의미하므로 answer을 1씩 늘림. 외부 loop 탈출 후, 최종적으로 저장된 answer을 return

1. vector내에 해당 원소가 존재하는지 확인
=> find(v.begin(), v.end(), 찾을 대상)
1) return 값이 v.end()인 경우 => 해당 원소가 존재하지 않음
2) return 값이 v.end()가 아닌 경우 => 해당 원소가 존재함
2. vector내에서 해당 원소가 위치하는 인덱스 찾기
=> find(v.begin(), v.end(), 찾을 대상) - v.begin

풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

int solution(vector<string> want, vector<int> number, vector<string> discount) {
    int answer = 0;
    vector<int> original_number = number;
    
    for(int i = 0; i <= discount.size() - 10; ++i) {
        int check = 1;
        number = original_number;
        for(int j = i; j < i + 10; ++j) {
            auto found = find(want.begin(), want.end(), discount[j]);
            if(found != want.end() 
            && number[found - want.begin()] != 0) {
                number[found - want.begin()]--;
            }
            else if(found == want.end() 
            || number[found - want.begin()] == 0) {
                check = 0;
                break;
            }
        }
        if(check == 1) answer++;
    }
    return answer;
}