[USACO23DEC] Candy Cane Feast B

题目描述

Farmer John 的奶牛对甜食情有独钟，它们尤其喜欢吃糖果棒。FJ 共有 N头奶牛，每头奶牛都有一个特定的初始高度。他想要喂它们 M 根糖果棒，每根糖果棒的高度也各不相同（1<=N, M <= 2*10^5）。

FJ 计划按照输入给出的顺序，逐一喂给奶牛们糖果棒。然后，奶牛们会按照输入给出的顺序一个接一个地排队，走向糖果棒，每头奶牛最多吃到与它高度相同的部分（因为它们够不到更高的地方）。即使奶牛吃掉了糖果棒的底部，糖果棒也在最初悬挂的地方保持不动，并不会被降低到地面。如果糖果棒的底部已经高于某头奶牛的高度，那么这头奶牛在它的回合中可能什么也吃不到。每头奶牛轮流吃过后，它们的身高会增加它们吃掉的糖果棒的单位数量，然后农夫约翰挂上下一根糖果棒，奶牛们再次重复这个过程（第一头奶牛再次成为第一个开始吃下一根糖果棒的）。

输入格式

第一行包含 N和 M。

接下来的一行包含 N头奶牛的初始高度，每头奶牛的高度都在 [1,10^9] 的范围内。

接下来的一行包含 M 根糖果棒的长度，每根糖果棒的长度都在 [1,10^9] 的范围内。

输出格式

输出 N 行，表示每头奶牛最终的高度。

请注意，由于这个问题涉及的整数大小较大，可能需要使用 64 位整数数据类型（例如，在 C/C++ 中使用 `long long` 类型）。

 样例输入 
3 2
3 2 5
6 1

样例输出
7
2
7

样例解释

第一根糖果棒高度为 6 单位。

- 第一头奶牛吃掉了第一根糖果棒直至高度  3 的部分，之后第一根糖果棒剩余高度 [3,6] 的部分。
- 第二头奶牛不够高，无法吃掉第一根糖果棒的任何剩余部分。
- 第三头奶牛额外吃掉了第一根糖果棒的两个单位。第一根糖果棒的剩余高度 [5,6] 的部分未被吃掉。

接下来，每头奶牛根据它吃掉的数量增长，所以奶牛的高度变为 [3+3, 2+0, 5+2]=[6, 2, 7]。

第二根糖果棒高度为 1 单位，被第一头奶牛全部吃掉。

题目解析

举一个例子：

#include <iostream>
using namespace std;
const int N = 1e5 + 10;
long long h[N], s[N];

int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> h[i]; //h[i]表示奶牛身高
    }
    for (int i = 1; i <= m; i++) {
        cin >> s[i]; //s[i]表示每串糖果个数
    }
       for (int i = 1; i <= m; i++) { //一串一串喂糖果
        int dist = 0; //初始化糖果距离地面的距离，因为一开始都是能接触地面的，所以为0
        int eaten = 0; //初始化可以吃几颗糖果
        for (int j = 1; j <= n; j++) { //奶牛排队吃糖果
            if (h[j] >= dist) { //奶牛的身高可以吃到糖果
                if (s[i]+dist < h[j]) { //奶牛的身高超过了剩余的糖果数量
                    eaten = s[i]; //可以吃的糖果数就是剩余的糖果数
                }
                else {
                    eaten = h[j] - dist; //否则就是吃到身高最高的那个位置
                }
                h[j] = h[j] + eaten; //更新奶牛身高
                s[i] = s[i] - eaten; //更新糖果数
                dist = dist + eaten; //糖果吃了之后距离地面变远了
            }
            else eaten = eaten + 0; 
            if (s[i] == 0) { //如果有奶牛吃完了所有糖果
                break; //跳出当前循环
            }
        }
    }
    for (int j = 1; j <= n; j++) {
        printf("%d \n", h[j]);
    }
}

方法二：可以建立糖果棒目前可以吃的高度区间[l,r]。当牛的高度完全大于等于 r (即能够把糖果棒全部吃掉)，那就把目前糖果棒能吃的部分全部加到牛的高度上；如果牛的高度是在[l,r]中间部分的，那么牛能吃的部分就是牛的高度a[j]减去l。枚举的过程中需要维护l。（r一开始就定好了，没必要去更改）

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
ll n, m, l, r, k, a[200005];

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    for (int i = 1; i <= m; i++) {
        cin >> r;
        l = 1;
        for (int j = 1; j <= n; j++) {
            if (a[j] >= r) {
                a[j] += r - l + 1;
                break;
            }
            else if (a[j] >= l) {
                k = a[j];
                a[j] += a[j] - l + 1;
                l = k + 1;
            }
        }
    }
    for (int i = 1; i <= n; i++)
        cout << a[i] << endl;
    return 0;
}