算法训练 石子游戏 python

算法训练 石子游戏

问题描述
　　石子游戏的规则如下：
　　地上有n堆石子，每次操作可选取两堆石子（石子个数分别为x和y）并将它们合并，操作的得分记为(x+1)×(y+1)，对地上的石子堆进行操作直到只剩下一堆石子时停止游戏。
　　请问在整个游戏过程中操作的总得分的最大值是多少？

输入格式
　　输入数据的第一行为整数n，表示地上的石子堆数；第二行至第n+1行是每堆石子的个数。
　　
输出格式
　　程序输出一行，为游戏总得分的最大值。
　　
样例输入
10
5105
19400
27309
19892
27814
25129
19272
12517
25419
4053

样例输出
15212676150

数据规模和约定
1≤n≤1000，1≤一堆中石子数≤50000

思路：
运用贪心算法思想，每次都取石子数量最多和第二多的两堆石子进行合并操作（进行排序操作后取前两个数），即可得到每次操作的得分为最大，最后再将每次的操作得分最大值相加求和即可得到整个游戏过程中操作的总得分的最大值。

代码：

n = int(input())
stones = []
totals = []
for i in range(n):
    stones.append(int(input()))
total = 1
stones.sort(reverse = True)

for j in range(n-1):
    if len(stones) == 1:
        break
    total = (stones[0]+1)*(stones[1]+1)
    totals.append(total)
    stones.insert(2, stones[0]+stones[1])
    del stones[:2]
    
print(sum(totals))

欢迎大家采纳和指正！