LeetCode 1860. 增长的内存泄露（等差数列）

示例 1：
输入：memory1 = 2, memory2 = 2
输出：[3,1,0]
解释：内存分配如下：
- 第 1 秒，内存条 1 被占用 1 位内存。内存条 1 现在有 1 位剩余可用内存。
- 第 2 秒，内存条 2 被占用 2 位内存。内存条 2 现在有 0 位剩余可用内存。
- 第 3 秒，程序意外退出，两个内存条分别有 1 位和 0 位剩余可用内存。

示例 2：
输入：memory1 = 8, memory2 = 11
输出：[6,0,4]
解释：内存分配如下：
- 第 1 秒，内存条 2 被占用 1 位内存，内存条 2 现在有 10 位剩余可用内存。
- 第 2 秒，内存条 2 被占用 2 位内存，内存条 2 现在有 8 位剩余可用内存。
- 第 3 秒，内存条 1 被占用 3 位内存，内存条 1 现在有 5 位剩余可用内存。
- 第 4 秒，内存条 2 被占用 4 位内存，内存条 2 现在有 4 位剩余可用内存。
- 第 5 秒，内存条 1 被占用 5 位内存，内存条 1 现在有 0 位剩余可用内存。
- 第 6 秒，程序意外退出，两个内存条分别有 0 位和 4 位剩余可用内存。
 
提示：
0 <= memory1, memory2 <= 2^31 - 1

class Solution {
public:
    vector<int> memLeak(int memory1, int memory2) {
        int t = 1;
        while(memory1 >= t || memory2 >= t)
        {
            if(memory1 == memory2)//相等的情况
            {
                memory1 -= t;
                t++;
            }
            else
            {
                bool _1islarge = memory1 > memory2;
                int dis = abs(memory1 - memory2); // 差距
                long long delta = (sqrt(4LL*t*t-4LL*t+1+8LL*dis)-2LL*t-1)/2;
                // 等差数列 t+(t+1)+...+(t+delta) = dis, 求 delta 
                int used = (delta+1)*(2*t+delta)/2;
                t += delta+1;
                if(_1islarge)
                    memory1 -= used;
                else
                    memory2 -= used;
            }
        }
        return {t, memory1, memory2};
    }
};