青训营X豆包MarsCode 技术训练营第四课 ｜ 豆包MarsCode AI 刷题

问题描述

小R有nn部电脑，每部电脑的电池容量分别为aiai​。她可以使用两种不同的充电方式来给电脑充电：

1. 普通充电：每单位时间为电脑充电xx单位的电量。
2. 闪充：每单位时间为电脑充电4x4x单位的电量。

现在，所有电脑的电量都为零。小R希望使用闪充给所有电脑充满电，计算她需要的总充电时间。请保留结果的小数点后两位。

测试样例

样例1：

输入：n = 4 ,x = 1 ,a = [2, 3, 4, 5]
输出：'3.50'

样例2：

输入：n = 3 ,x = 2 ,a = [4, 6, 8]
输出：'2.25'

样例3：

输入：n = 2 ,x = 1 ,a = [10, 5]
输出：'3.75'

要解决这个问题，我们需要计算每部电脑使用闪充充满电所需的时间，并求出总时间。具体步骤如下：

1. 计算每部电脑的充电时间：对于每部电脑，使用闪充充电的时间为 a_i / (4 * x)。
2. 求和：将所有电脑的充电时间求和，得到总充电时间。
3. 格式化输出：将总充电时间保留两位小数并输出。
4. 1. 计算每部电脑的充电时间：对于每部电脑，使用闪充充电的时间为 a_i / (4 * x)。
5. 求和：将所有电脑的充电时间求和，得到总充电时间。
6. 格式化输出：将总充电时间保留两位小数并输出。

复杂度分析：

• 时间复杂度：O(n)，其中 n 是电脑的数量。我们只需要遍历电池容量数组一次。
• 空间复杂度：O(1)，只使用了常数空间。

这个实现能够正确计算使用闪充给所有电脑充满电所需的总充电时间，并且在合理的时间内处理较小的输入。对于较大的输入，可能需要进一步优化。

def solution(n: int, x: int, a: list) -> str:
    # 计算每台电脑充满电所需的时间
    times = [ai / (4 * x) for ai in a]
    
    # 计算总时间
    total_time = sum(times)
    
    # 保留两位小数并返回
    return f"{total_time:.2f}"
     
    
    

    
    # 保留两位小数并返回
    return f"{times:.2f}"

if __name__ == '__main__':
    print(solution(4, 1, [2, 3, 4, 5]) == '3.50')
    print(solution(3, 2, [4, 6, 8]) == '2.25')
    print(solution(2, 1, [10, 5]) == '3.75')