[Javascript] 실전 대비 모의고사 1차 2번

 

테스트 케이스
want = ["banana", "apple", "rice", "pork", "pot"]

number = [3,2,2,2,1]

discount = ["chicken", "apple", "apple", "banana", "rice", "apple", "pork", "banana", "pork", "rice", "pot", "banana", "apple", "banana"]

XYZ 마트는 일정 금액을 지불하면 10일간 회원 자격 부여한다.
회원을 대상으로는 매일 한가지 제품을 할인하고 할인 제품은 하루에 1개만 구매 가능하다.

정현이는 원하는 제품과 수량이 할인하는 날짜와 10일 연속으로 일치할 경우에 맞춰 회원가입하길 원한다.
돼지고기 5개, 바나나 5개를 원할 경우 할인 범위에 돼지고기 5번, 바나나 5번이 포함되어야 한다.
=> discount = ["pork", "pork", "pork", "pork", "pork", "banana", "banana", "banana", "banana", "banana"] 인 경우가 있어야 한다. (순서는 상관없음)

접근 방법
sliding window 방식으로 discount의 모든 범위를 탐색하고 O(N)
가능한 경우 answer += 1하는 방식으로 접근하였다.

만약 11일인 경우 0일차를 빼고 11일차를 더하고
12일인 경우 1일차를 빼고 12일차를 더하는 방식으로 진행

 우선 물건을 뺄 날짜를 저장하는 prev와 물건을 더할 날짜를 저장하는 next 변수를 만들고,
정현이가 원하는 물건과 개수를 key와 value로 저장하는 객체를 생성하였다.

let prev = 0;
let next = 0;

const shoppingObj = {};
want.forEach((e,idx) => shoppingObj[e] = number[idx]);​

결과로 shoppingObj = {banana: 3, apple: 2, rice: 2, pork: 2, pot: 1}이 생성되고
이 shoppingObj의 모든 value가 0이 되면 모든 제품을 할인 받는 날이 된다.


이제 discount를 순회하면서
1. 다음날(next) shoppingObj[key] === discount 일때 -1; 
2. 지나간 날(prev) shoppingObj[key] === discount 일때 +1;
3. shoppingObj[key] !== discount 일때 아무것도 하지 않는다.
라는 로직을 진행한다.


 우선 첫 10일간의 discount 값을 shoppingObj에 반영한다.

while(next < 10){
    const item = discount[next];
    if(!Object.keys(shoppingObj).includes(item)){
        next++;
    }else{
        shoppingObj[item] -= 1;
        next++;
    }
}

다음으로 앞서 살핀 로직대로 next가 discount의 마지막 날까지 진행한다.

 while(next < discount.length){
    let addItem = discount[next];
    let delItem = discount[prev];

    if(!Object.keys(shoppingObj).includes(addItem)){
        next++;
    }else{
        shoppingObj[addItem] -= 1;
        next++;
    }

    if(!Object.keys(shoppingObj).includes(delItem)){
        prev++;
    }else{
        shoppingObj[delItem] += 1;
        prev++;
    }

    if(allItemSale(shoppingObj)) answer += 1;
}

allItemSale은 object를 받아와 object의 모든 값이 0이면 true를 반환하여 모든 값이 0인지 체크한다.

if(allItemSale(shoppingObj)) answer += 1;

const allItemSale = (obj) => {
    let result = Object.values(obj).filter(e => e !== 0);
    return result.length === 0 ? true : false;
}

고칠점

1. 굳이 discount 10까지 따로 처리해야 할까?

while(next < 10){
    const item = discount[next];
    if(!Object.keys(shoppingObj).includes(item)){
        next++;
    }else{
        shoppingObj[item] -= 1;
        next++;
    }
}

2. 이런 형식의 코드가 3번이나 등장한다. 

if(!Object.keys(shoppingObj).includes(addItem)){
    next++;
}else{
    shoppingObj[addItem] -= 1;
    next++;
}

코드 수정

if(next - prev < 10){
    next = setAddItem(shoppingObj, addItem, next);
}else{
    next = setAddItem(shoppingObj, addItem, next);
    prev = setDelItem(shoppingObj, delItem, prev);
}

const setAddItem = (shoppingObj, item, index) => {
    if(Object.keys(shoppingObj).includes(item)){
        shoppingObj[item] -= 1;
    }
    index += 1;
    return index;
}

const setDelItem = (shoppingObj, item, index) => {
    if(Object.keys(shoppingObj).includes(item)){
        shoppingObj[item] += 1;
    }
    index += 1;
    return index;
}

1. if(next > prev)일때를 본 로직에 추가하여 해결

2. 함수를 만들어 정리

하나의 setItem으로 만들고 싶었는데
shoppingObj[item] -= 1;
shoppingObj[item] += 1;
두 조건을 처리하기 어려워 두가지 함수로 작성하였다.

 

function solution(want, number, discount) {
    var answer = 0;
    let prev = 0;
    let next = 0;
    
    const shoppingObj = {};
    want.forEach((e,idx) => shoppingObj[e] = number[idx]);

    while(next - prev < 10){
        const item = discount[next];
        if(!Object.keys(shoppingObj).includes(item)){
            next++;
        }else{
            shoppingObj[item] -= 1;
            next++;
        }
    }

    if(allItemSale(shoppingObj)) answer += 1;

    while(next < discount.length){
        let addItem = discount[next];
        let delItem = discount[prev];

        if(!Object.keys(shoppingObj).includes(addItem)){
            next++;
        }else{
            shoppingObj[addItem] -= 1;
            next++;
        }

        if(!Object.keys(shoppingObj).includes(delItem)){
            prev++;
        }else{
            shoppingObj[delItem] += 1;
            prev++;
        }

        if(allItemSale(shoppingObj)) answer += 1;
    }

    return answer;
}

const allItemSale = (obj) => {
    let result = Object.values(obj).filter(e => e !== 0);
    return result.length === 0 ? true : false;
}

// 리팩토링 코드

function solution(want, number, discount) {
    var answer = 0;
    let prev = 0;
    let next = 0;    
    const shoppingObj = {};
    
    want.forEach((e,idx) => shoppingObj[e] = number[idx]);

    while(next < discount.length){
        let addItem = discount[next];
        let delItem = discount[prev];

        if(next - prev < 10){
            next = setAddItem(shoppingObj, addItem, next);
        }else{
            next = setAddItem(shoppingObj, addItem, next);
            prev = setDelItem(shoppingObj, delItem, prev);
        }

        if(isAllItemSale(shoppingObj)) answer += 1;
    }

    return answer;
}

const setAddItem = (shoppingObj, item, index) => {
    if(Object.keys(shoppingObj).includes(item)){
        shoppingObj[item] -= 1;
    }
    index += 1;
    return index;
}

const setDelItem = (shoppingObj, item, index) => {
    if(Object.keys(shoppingObj).includes(item)){
        shoppingObj[item] += 1;
    }
    index += 1;
    return index;
}
    
const isAllItemSale = (obj) => {
    let result = Object.values(obj).filter(e => e !== 0);
    return result.length === 0 ? true : false;
}