[LeetCode] Maximum Units On A Truck

문제링크

Maximum Units on a Truck - LeetCode

 

boxTypes에는 [박스 갯수, 박스당 담을 수 있는 물건의 수]가 들어있고
truckSize에는 최대 실을 수 있는 박스의 갯수가 주어진다.

이때 트럭에 담을 수 있는 최대 물건의 수를 구하라는 문제이다.

접근 방법

최대 실을 수 있는 물건은 "박스 당 담을 수 있는 물건의 수"가 가장 많은 박스들을 우선으로 담는 것 이다.

그러므로 박스의 물건의 수가 많은 순으로 내림차 순으로 정렬한다. 

boxTypes.sort((a,b) => b[1] - a[1])​

그리고 현재 트럭에 들어있는 박스의 수를 의미하는 boxCnt 변수를 생성하여
1. 현재 트럭의 박스 + 앞으로 들어갈 박스 < 트럭에 실을 수 있는 박스 수
2. 현재 트럭의 박스 + 앞으로 들어갈 박스 >= 트럭에 실을 수 있는 박스 수

두 가지 경우를 조건문으로 만들어 문제를 해결하였다.

if((boxCnt + box) < truckSize){
  output += box * items;
  boxCnt += box;
}else{
  output += (truckSize - boxCnt) * items;
  boxCnt += (truckSize - boxCnt);
  return false;
}​

주의사항

forEach문은 break를 사용할 수 없다. 
때문에 유사한 방법으로 return false;를 사용할 수 있다.

 

/**
 * @param {number[][]} boxTypes
 * @param {number} truckSize
 * @return {number}
 */
 var maximumUnits = function(boxTypes, truckSize) {
  let output = 0;
  let boxCnt = 0;
  
  boxTypes.sort((a,b) => b[1] - a[1])

  boxTypes.forEach(boxType => {
      let [box, items] = boxType;
      if((boxCnt + box) < truckSize){
          output += box * items;
          boxCnt += box;
      }else{
          output += (truckSize - boxCnt) * items;
          boxCnt += (truckSize - boxCnt);
          return false;
      }
  })
  
  return output;   
};