[알고리즘 || 프로그래머스] 카드 뭉치

function solution(cards1, cards2, goal) {
    let rmCount = 0;
    goal.map((word) => {
        if (cards1[0] === word) {
            rmCount += 1;
            cards1.splice(0, 1);
        } else if (cards2[0] === word) {
            rmCount += 1;
            cards2.splice(0, 1);
        } else {
            return;
        }
    })
    return rmCount === goal.length ? "Yes" : "No"
}

 

첨에 이 조건이 있는 줄 모르고, 

cards1, 2 의 length가 0 이면 yes 를 return 했다. 

 

 

아래는 비난님이 위의 코드를 중요도에 따라 반복문을 없애는 방법으로 제안해주신 코드

function solution(cards1, cards2, goal) {
    let rmCount = 0;
    goal.map((word) => {
        if (word === cards1[0] || word === cards2[0]) {
            rmCount += 1;
        } else {
            return;
        }

        if (cards1[0] === word) {
            cards1.splice(0, 1);
        } else {
            cards2.splice(0, 1);
        }
    })
    return rmCount === goal.length ? "Yes" : "No"
}

=> rmCount += 1 가 한번만 들어간다. 

=> || 연산자 사용으로 '둘 중 하나에 포함될 경우' 를 나타냈다. 포함되어 있지 않으면 return 을 해준다. 

=> 포함되어 있으면 splice(0, 1) 으로 cards 배열 자체를 변형한다. (맨 앞 1개만 제거)

마지막은 이전 코드처럼 제거한 count 가 goal 의 length 와 같다면 Yes 를 return 한다. 

 

 

 

 

+ 사실 length 에 집착하지 않아도 된다. 

map 을 사용하지 않고 for 문을 사용한다면 바로 return 을 해줄 수 있다. 

 

function solution(cards1, cards2, goal) {
    for(const word of goal){
        if(word !== cards1[0] && word !== cards2[0]) return "No"
        if(word === cards1[0]) cards1.shift();
        if(word === cards2[0]) cards2.shift();
    }
    
    return "Yes";
}

 

 

every 와 shift 를 사용하면 이렇게 한 줄로도 가능하다. 

const solution = (cards1, cards2, goal) => goal.every(word => word === cards1[0] ? cards1.shift() : word === cards2[0] ? cards2.shift() : false) ? "Yes" : "No";