Question

1860. Incremental Memory Leak

中文文档

Description

You are given two integers memory1 and memory2 representing the available memory in bits on two memory sticks. There is currently a faulty program running that consumes an increasing amount of memory every second.

At the ith second (starting from 1), i bits of memory are allocated to the stick with more available memory (or from the first memory stick if both have the same available memory). If neither stick has at least i bits of available memory, the program crashes.

Return _an array containing _[crashTime, memory1crash, memory2crash]_, where _crashTime_ is the time (in seconds) when the program crashed and _memory1crash_ and _memory2crash_ are the available bits of memory in the first and second sticks respectively_.

 

Example 1:

Input: memory1 = 2, memory2 = 2
Output: [3,1,0]
Explanation: The memory is allocated as follows:
- At the 1st second, 1 bit of memory is allocated to stick 1. The first stick now has 1 bit of available memory.
- At the 2nd second, 2 bits of memory are allocated to stick 2. The second stick now has 0 bits of available memory.
- At the 3rd second, the program crashes. The sticks have 1 and 0 bits available respectively.

Example 2:

Input: memory1 = 8, memory2 = 11
Output: [6,0,4]
Explanation: The memory is allocated as follows:
- At the 1st second, 1 bit of memory is allocated to stick 2. The second stick now has 10 bit of available memory.
- At the 2nd second, 2 bits of memory are allocated to stick 2. The second stick now has 8 bits of available memory.
- At the 3rd second, 3 bits of memory are allocated to stick 1. The first stick now has 5 bits of available memory.
- At the 4th second, 4 bits of memory are allocated to stick 2. The second stick now has 4 bits of available memory.
- At the 5th second, 5 bits of memory are allocated to stick 1. The first stick now has 0 bits of available memory.
- At the 6th second, the program crashes. The sticks have 0 and 4 bits available respectively.

 

Constraints:

• 0 <= memory1, memory2 <= 231 - 1

Solutions

Solution 1: Simulation

We directly simulate the allocation of memory.

Assume $t$ is the moment of unexpected exit, then the two memory sticks can definitely accommodate the memory consumed at the moment $t-1$ and before, so we have:

$$ \sum_{i=1}^{t-1} i = \frac{t\times (t-1)}{2} \leq (m_1+m_2) $$

The time complexity is $O(\sqrt{m_1+m_2})$, where $m_1$ and $m_2$ are the sizes of the two memory sticks respectively.

class Solution:
  def memLeak(self, memory1: int, memory2: int) -> List[int]:
    i = 1
    while i <= max(memory1, memory2):
      if memory1 >= memory2:
        memory1 -= i
      else:
        memory2 -= i
      i += 1
    return [i, memory1, memory2]

class Solution {
  public int[] memLeak(int memory1, int memory2) {
    int i = 1;
    for (; i <= Math.max(memory1, memory2); ++i) {
      if (memory1 >= memory2) {
        memory1 -= i;
      } else {
        memory2 -= i;
      }
    }
    return new int[] {i, memory1, memory2};
  }
}

class Solution {
public:
  vector<int> memLeak(int memory1, int memory2) {
    int i = 1;
    for (; i <= max(memory1, memory2); ++i) {
      if (memory1 >= memory2) {
        memory1 -= i;
      } else {
        memory2 -= i;
      }
    }
    return {i, memory1, memory2};
  }
};

func memLeak(memory1 int, memory2 int) []int {
  i := 1
  for ; i <= memory1 || i <= memory2; i++ {
    if memory1 >= memory2 {
      memory1 -= i
    } else {
      memory2 -= i
    }
  }
  return []int{i, memory1, memory2}
}

function memLeak(memory1: number, memory2: number): number[] {
  let i = 1;
  for (; i <= Math.max(memory1, memory2); ++i) {
    if (memory1 >= memory2) {
      memory1 -= i;
    } else {
      memory2 -= i;
    }
  }
  return [i, memory1, memory2];
}

/**
 * @param {number} memory1
 * @param {number} memory2
 * @return {number[]}
 */
var memLeak = function (memory1, memory2) {
  let i = 1;
  for (; i <= Math.max(memory1, memory2); ++i) {
    if (memory1 >= memory2) {
      memory1 -= i;
    } else {
      memory2 -= i;
    }
  }
  return [i, memory1, memory2];
};