[프로그래머스 Lv.2] 할인 행사

코딩테스트 연습 - 할인 행사

문제 설명

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

풀이

To-Do 리스트

1. 구매 총 갯수를 구하기
2. 첫째날부터 10일간의 할인 목록을 구함.
   2-1. 이때, n번째날+10 보다 할인 목록의 길이가 적어지는 경우에는 할인 목록까지로 지정.
   2-2. 해당 품목의 갯수를 구해 map에 저장.
   2-3. 할인 목록을 구할때 구매 총 갯수보다 적어지는 경우에는 반복을 멈춤.
3. 해당 품목이 없거나 해당 품목에 대한 구매 갯수가 맞지 않는 경우에는 반복을 멈춤.
4. 품목갯수가 맞으면, answer에 1을 더한다.

결과

import java.util.*;
import java.util.stream.*;

class Solution {
    public int solution(String[] want, int[] number, String[] discount) {
        int answer = 0;
        // 일정한 금액을 지불하면 10일 동안 회원 자격을 부여
        // 할인하는 제품은 하루에 하나씩만 구매
        // 자신이 원하는 제품과 수량이 할인하는 날짜, 10일 연속으로 일치할 경우에 맞춰서 회원가입
        int tot = 0;
        for(int n : number){
            tot += n;
        }

        int day = 0;
        while(true){

            int len = discount.length < day+10 ? discount.length : day+10;
            List<String> pur = new ArrayList<>();
            Map<String,Integer> map = new HashMap<>();
            
            for(int j = day; j < len; j++){
                pur.add(discount[j]);
                map.put(discount[j], map.getOrDefault(discount[j],0)+1);
            }
            if(pur.size() < tot) break;
            
            int cnt = 0;
            for(int i = 0; i < want.length; i++){
                String prod = want[i];
                int amount = number[i];               
                if(map.get(prod) == null || map.get(prod) != amount){
                    break;
                }
                cnt++;
            }
            if(cnt == number.length) answer++;
            
            day++;   
        }
        return answer;
    }
}