[LeetCode] Maximum Bags With Full Capacity Of Rocks

문제링크

Maximum Bags With Full Capacity of Rocks - LeetCode

 

돌을 담을 수 있는 용량 : capacity
현재 담겨있는 돌 : rocks
추가로 담을 수 있는 돌의 개수 : additionalRocks

위의 값들이 주어졌을때 "현재 담겨있는 돌"에서 돌을 더 추가해서 "돌을 담을 수 있는 용량"을 최대한 채우고, 이때 모두 채워진 capacity의 개수를 return 하여라.

[2,3,4,5]라는 capacity가 주어지고, [1,2,4,5]라는 현재 담긴 돌이 주어졌을때
additionalRocks = 1 이라면 [2,2,4,5] 또는 [1,3,4,5]를 만들 수 있고, 모두 가득찬 capacity는 3개 이므로 3을 return 한다.

접근 방법

1. 주어진 배열 capacity와 rocks를 서로 뺀 배열을 생성한다. (모자란 돌의 개수를 의미하는 배열)

let needRocks = capacity.map((rock, idx) => {
  return rock - rocks[idx];
});​

2. needRocks를 오름차순으로 정렬하고, 조금 필요한 돌부터 채우는 방향으로 진행한다.

needRocks.sort((a, b) => a - b);

3. 필요한 돌이 0개면 output++;

    필요한 돌이 1개 이상이라면 additionalRocks에 필요한 돌을 뺀다.

needRocks.forEach(rock => {
  if(rock === 0) output++;
  else{
      additionalRocks -= rock;
      if(additionalRocks >= 0) output++;
      else return false;
  }
})​

필요한 돌이 1개 이상이라서 additionalRocks에서 필요한 돌을 뺏을 때,
addtionalRocks가 0개 이상이여야 output++을 하고, 아니라면 반복문을 멈춘다. 

 

/**
 * @param {number[]} capacity
 * @param {number[]} rocks
 * @param {number} additionalRocks
 * @return {number}
 */
 var maximumBags = function(capacity, rocks, additionalRocks) {
  let output = 0;
  let needRocks = capacity.map((rock, idx) => {
      return rock - rocks[idx];
  });
  
  needRocks.sort((a, b) => a - b);
  
  needRocks.forEach(rock => {
      if(rock === 0) output++;
      else{
          additionalRocks -= rock;
          if(additionalRocks >= 0) output++;
          else return false;
      }
  })
  
  return output;
};