1. 问题描述:
一共有 n 个箱子排成一排,从左到右依次编号为 1∼n。其中,第 i 号箱子中放有 ai 个石子。现在,你可以进行最多 d 次操作。每次操作可以将一个石子从一个箱子移动至另一个与其相邻的箱子里。我们希望通过合理操作使得 1 号箱子内的石子数量尽可能大。请问,这个最大可能值是多少?
输入格式
第一行包含整数 T,表示共有 T 组测试数据。每组数据第一行包含两个整数 n 和 d。第二行包含 n 个整数 a1,a2,…,an。
输出格式
每组数据输出一行结果,表示答案。
数据范围
1 ≤ T ≤ 100,
1 ≤ n,d ≤ 100,
0 ≤ ai ≤ 100
输入样例:
3
4 5
1 0 3 2
2 2
100 1
1 8
0
输出样例:
3
101
0
来源:https://www.acwing.com/problem/content/3772/
2. 思路分析:
分析题目可以知道我们要想使得1号箱子的石头最多,那么尽可能将靠近1号箱子的石头移动到1号箱子,如果能够移动就移动。
3. 代码如下:
class Solution:
def process(self):
T = int(input())
for t in range(T):
n, d = map(int, input().split())
a = list(map(int, input().split()))
res = 0
for i in range(n):
if i == 0:
res += a[i]
else:
# d // i表示一次能够移动多少个石头
t = min(d // i, a[i])
res += t
d -= t * i
print(res)
if __name__ == '__main__':
Solution().process()