LeetCdoe 2960. 统计已测试设备

2960. 统计已测试设备

给你一个长度为 n 、下标从 0 开始的整数数组 batteryPercentages ，表示 n 个设备的电池百分比。

你的任务是按照顺序测试每个设备 i，执行以下测试操作：

• 如果 batteryPercentages[i] 大于 0：
  • 增加 已测试设备的计数。
  • 将下标在 [i + 1, n - 1] 的所有设备的电池百分比减少 1，确保它们的电池百分比 不会低于 0 ，即 batteryPercentages[j] = max(0, batteryPercentages[j] - 1)。
  • 移动到下一个设备。
• 否则，移动到下一个设备而不执行任何测试。

返回一个整数，表示按顺序执行测试操作后 已测试设备 的数量。

示例 1：

输入：batteryPercentages = [1,1,2,1,3]
输出：3
解释：按顺序从设备 0 开始执行测试操作：
在设备 0 上，batteryPercentages[0] > 0 ，现在有 1 个已测试设备，batteryPercentages 变为 [1,0,1,0,2] 。
在设备 1 上，batteryPercentages[1] == 0 ，移动到下一个设备而不进行测试。
在设备 2 上，batteryPercentages[2] > 0 ，现在有 2 个已测试设备，batteryPercentages 变为 [1,0,1,0,1] 。
在设备 3 上，batteryPercentages[3] == 0 ，移动到下一个设备而不进行测试。
在设备 4 上，batteryPercentages[4] > 0 ，现在有 3 个已测试设备，batteryPercentages 保持不变。
因此，答案是 3 。

示例 2：

输入：batteryPercentages = [0,1,2]
输出：2
解释：按顺序从设备 0 开始执行测试操作：
在设备 0 上，batteryPercentages[0] == 0 ，移动到下一个设备而不进行测试。
在设备 1 上，batteryPercentages[1] > 0 ，现在有 1 个已测试设备，batteryPercentages 变为 [0,1,1] 。
在设备 2 上，batteryPercentages[2] > 0 ，现在有 2 个已测试设备，batteryPercentages 保持不变。
因此，答案是 2 。

提示：

• 1 <= n == batteryPercentages.length <= 100
• 0 <= batteryPercentages[i] <= 100

---

提示 1

One solution is simulating the operations as explained in the problem statement, and it works in O(n^2) time.

---

提示 2

While going through the devices, you can maintain the number of previously tested devices, and the current device can be tested if batteryPercentages[i] is greater than the number of tested devices.

解法1：差分数组

我们可以利用「差分数组」的思想，用 count 表示当前元素需要减 1 的次数，如果当前元素 x=batteryPercentages[i]，如果 x>count 时，当前元素实际的值为 x−count>0，i 后面的所有元素都需要减 1，此时需对 count 加 1，当遍历完数组后，返回 count 即可。

Java版：

class Solution {
    public int countTestedDevices(int[] batteryPercentages) {
        int n = batteryPercentages.length;
        int[] diff = new int[n];
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (batteryPercentages[i] > count) {
                count++;
            }
        } 
        return count;
    }
}

Python3版：

class Solution:
    def countTestedDevices(self, batteryPercentages: List[int]) -> int:
        n = len(batteryPercentages)
        count = 0
        for p in batteryPercentages:
            if p > count:
                count += 1
        return count        

复杂度分析

• 时间复杂度：O(n), n 表示给定的数组 batteryPercentages 的长度。只需遍历一遍数组即可。

• 空间复杂度：O(1)。