AcWing 3769. 移动石子

问题描述

一共有$n$个箱子排成一排，从左到右依次编号为$1\sim n$。

其中，第$i$号箱子中放有$ai$个石子。

现在，你可以进行最多$d$次操作。

每次操作可以将一个石子从一个箱子移动至另一个与其相邻的箱子里。

我们希望通过合理操作使得$1$号箱子内的石子数量尽可能大。

请问，这个最大可能值是多少？

输入格式：

第一行包含整数$T$，表示共有$T$组测试数据。

每组数据第一行包含两个整数$n$和$d$。

第二行包含$n$个整数$a1,a2,\dots ,an$。

输出格式：

每组数据输出一行结果，表示答案。

数据范围

$1\le T\le 100,$
$1\le n,d\le 100,$
$0\le ai\le 100$

输入样例：

3
4 5
1 0 3 2
2 2
100 1
1 8
0

输出样例：

3
101
0

思路

原题链接

C++代码：

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

int main()
{
    int T;
    cin >> T;
    while (T -- )
    {
        int n, d;
        cin >> n >> d;
        int res = 0;
        //从前往后枚举每个箱子
        for (int i = 0; i < n; i ++ )
        {
            int x;
            cin >> x; //读入当前箱子里的数量
            if (!i) res += x; //当前箱子就是第一个箱子
            else
            {
                int t = min(x, d / i); //能往第一堆移动多少个 d / i 下取整 最多移动 d/ i次
                res += t;
                d -= t * i; //操作次数
            }
        }
        cout << res << endl;
    }

    return 0;
}

#include <iostream>

using namespace std;

const int N = 1010;

int a[N];

int main()
{
    int T;
    cin >> T;

    while (T -- )
    {
        int n, d;
        cin >> n >> d;

        for (int i = 1; i <= n; i ++ ) cin >> a[i];

        //从第二个箱子开始前往后枚举箱子
        for (int i = 2; i <= n; i ++ )
        {
            while(a[i] != 0 && d >= (i - 1))
            {
                a[i] --;
                a[1] += 1;
                d -= (i - 1);
            }
        }
        cout << a[1] << endl;
    }
    return 0;   
}