掌上游戏机电池使用时间最大化算法 - Python 实现

掌上游戏机电池使用时间最大化算法 - Python 实现

小S新买了一个掌上游戏机，这个游戏机由两节5号电池供电。为了保证能够长时间玩游戏，他买了很多5号电池，这些电池的生产商不同，质量也有差异，因而使用寿命也有所不同，有的能使用5个小时，有的可能就只能使用3个小时。显然如果他只有两个电池一个能用5小时一个能用3小时，那么他只能玩3个小时的游戏，有一个电池剩下的电量无法使用，但是如果他有更多的电池，就可以更加充分地利用它们，比如他有三个电池分别能用3、3、5小时，他可以先使用两节能用3个小时的电池，使用半个小时后再把其中一个换成能使用5个小时的电池，两个半小时后再把剩下的一节电池换成刚才换下的电池（那个电池还能用2.5个小时），这样总共就可以使用5.5个小时，没有一点浪费。

现在已知电池的数量和电池能够使用的时间，请你找一种方案使得使用时间尽可能的长。

输入

输入包含多组数据。每组数据包括两行，第一行是一个整数N(2≤N≤1000)，表示电池的数目，接下来一行是N个正整数表示电池能使用的时间。

输出

对每组数据输出一行，表示电池能使用的时间，保留到小数点后1位。

输入样例

2
3 5
3
3 3 5

输出样例

3.0
5.5

解题思路

对于每组数据，首先将电池使用时间从小到大排序。然后，计算出每个电池的剩余使用时间，即当前电池的使用时间减去前面所有电池的使用时间之和。最后，选择剩余使用时间最长的两个电池使用，即排序后的最后两个电池。

代码实现

import sys

def max_game_time(N, battery_time):
    battery_time.sort()  # 将电池使用时间排序
    remain_time = [0] * N  # 记录每个电池的剩余使用时间
    for i in range(N):
        if i == 0:
            remain_time[i] = battery_time[i]
        else:
            remain_time[i] = battery_time[i] - sum(battery_time[:i])  # 计算剩余使用时间
    remain_time.sort()  # 将剩余使用时间排序
    max_time = remain_time[-1] + remain_time[-2]  # 选择剩余使用时间最长的两个电池使用
    return max_time

if __name__ == "__main__":
    for line in sys.stdin:
        N = int(line.strip())
        battery_time = list(map(int, sys.stdin.readline().strip().split()))
        max_time = max_game_time(N, battery_time)
        print("{:.1f}".format(max_time))