AcWing:3769. 移动石子

题目

一共有 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

思路简介 贪心

由题可知, 为了尽可能让第一个箱子的石头子多
我们要尽可能的移动距离第一个箱子最近的箱子里的石头子
再配合可移动的次数 进行判断

设 第 i 个箱子的石头子数量为 x, (第 i 个箱子到第 1 个箱子需要 i 次, i 从 0 开始)

可能出现的情况

1. 可移走第 i 个箱子里的全部石头子, 即 d / i > x
2. 只能移走第 i 个箱子里的部分石头子(d / i) 即 d / i < x

注意 以上除法皆为整除, 不能整除的小数位是不能到达第 1 个箱子的

• AC Code

#include <iostream>
#include <algorithm>
using namespace std;

int main () {
    int T; cin >> T;
    while (T--) {
        int n, d; cin >> n >> d;  // n 个箱子, d 次操作
        int res = 0;  // 记录第一个箱子的最大石头子数量
        for (int i = 0; i < n; i++) {
            int x; cin >> x;  // 第 i 个箱子石头子的数量
            if (i == 0) res += x;  // 第一次不需要移动, 直接全部加上
            else {
                int t = min(x, d / i);  // 判断 可移动石头子的数量
                res += t;  // 加上可移到 第 1 个箱子的石头子
                d -= t * i;  // 每次移动次数减去 t * i
            }
        }
        cout << res << "\n";
    }
}