Codeforces Round #621 (Div. 1 + Div. 2)_A. Cow and Haybales(C++ 시뮬레이션)

6417 단어
The USA Construction Operation (USACO) recently ordered Farmer John to arrange a row of n haybale piles on the farm. The i-th pile contains ai haybales.
However, Farmer John has just left for vacation, leaving Bessie all on her own. Every day, Bessie the naughty cow can choose to move one haybale in any pile to an adjacent pile. Formally, in one day she can choose any two indices i and j (1≤i,j≤n) such that |i−j|=1 and ai>0 and apply ai=ai−1, aj=aj+1. She may also decide to not do anything on some days because she is lazy.
Bessie wants to maximize the number of haybales in pile 1 (i.e. to maximize a1), and she only has d days to do so before Farmer John returns. Help her find the maximum number of haybales that may be in pile 1 if she acts optimally!
Input
The input consists of multiple test cases. The first line contains an integer t (1≤t≤100) — the number of test cases. Next 2t lines contain a description of test cases — two lines per test case.
The first line of each test case contains integers n and d (1≤n,d≤100) — the number of haybale piles and the number of days, respectively.
The second line of each test case contains n integers a1,a2,…,an (0≤ai≤100) — the number of haybales in each pile.
Output
For each test case, output one integer: the maximum number of haybales that may be in pile 1 after d days if Bessie acts optimally.
Example
input
3 4 5 1 0 3 2 2 2 100 1 1 8 0
output
3 101 0
Note
In the first test case of the sample, this is one possible way Bessie can end up with 3 haybales in pile 1:
On day one, move a haybale from pile 3 to pile 2 On day two, move a haybale from pile 3 to pile 2 On day three, move a haybale from pile 2 to pile 1 On day four, move a haybale from pile 2 to pile 1 On day five, do nothing In the second test case of the sample, Bessie can do nothing on the first day and move a haybale from pile 2 to pile 1 on the second day.
Process
This problem is easy. Logic can be wrong, but mathematics never.
Code
#include
using namespace std;
int t, n, d;
int main()
{
    cin >> t;
    for (int i = 0; i < t; i++)
    {
        int a[1000];
        cin >> n >> d;
        for (int i = 0; i < n; i++)
            cin >> a[i];
        for (int i = 1; i < n; i++)
        {
            int temp = min(d / i, a[i]);
            a[0] += temp;
            d -= (temp * i);
            if (d == 0)
                break;
        }
        cout << a[0] << endl;
    }
    return 0;
}

좋은 웹페이지 즐겨찾기