P2827 [NOIP2016 提高组] 队列

题意

传送门 P2827 [NOIP2016 提高组] 蚯蚓

题解

容易想到使用二叉堆维护长度的最大值，由于每次新产生之外的蚯蚓长度都增加 $q$，而难以修改已经在堆中的元素，那么对新产生的蚯蚓长度减 $q$，可以使长度的相互关系保持不变。时间复杂度 $O(m\log n)$，显然难以胜任。

设当前切割蚯蚓长度为 $x_1$，那么新增蚯蚓长度分别为 $\lfloor p{x}_1\rfloor$ 与 $x_1-\lfloor p{x}_1\rfloor$。那么下一次可能的切割蚯蚓显然为之前未切割的蚯蚓的长度最大值与新增蚯蚓长度，假设下一次切割的蚯蚓长度为 $x_2$，那么有 $x_1\ge x_2$，且满足

$$\{\begin{array}{l}\lfloor p{x}_1\rfloor +q=\lfloor p{x}_1+q\rfloor \ge \lfloor p(x_2+q)\rfloor \\ x_1-\lfloor p{x}_1\rfloor +q\ge x_1-\lfloor p(x_1+q)\rfloor +q\ge x_2-\lfloor p(x_2+q)\rfloor +q\end{array}$$ 第 $2$ 个不等式中 $x-\lfloor p(x+q)\rfloor ,x\in {\mathbb{Z}}^{+}$ 的单调性可通过将整数 $x$ 移入取整符号进行证明，已知 $0<1-p<1$，那么有
$$x_1-\lfloor p(x_1+q)\rfloor \ge \lfloor (1-p)x_1-pq)\rfloor \ge \lfloor (1-p)x_2-pq)\rfloor \ge x_2-\lfloor p(x_2+q)\rfloor$$ 证明了每一次新增的蚯蚓长度 $\lfloor p(x_2+q)\rfloor$ 与 $x_2-\lfloor p(x_2+q)\rfloor +q$ 分别不大于上一次切割新增的 $\lfloor p{x}_1\rfloor +q$ 与 $x_1-\lfloor p{x}_1\rfloor +q$。那么可以建立 $3$ 个队列，分别维护有序的初始值、新增的 $\lfloor p{x}_1\rfloor$ 部分与 $x_1-\lfloor p{x}_1\rfloor$ 部分，此时各队列保持有序，下一次切割长度为 $3$ 个队首元素的最大值。时间复杂度 $O(n\log n+m)$。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100005, maxm = 7000005, inf = 0x7fffffff;
int A[maxn], cut[maxm], rem[maxm + maxn];
queue<int> X, Y, Z;

int get()
{
    int x = X.size() ? X.front() : -inf, k = 0;
    if (Y.size() && Y.front() > x)
        x = Y.front(), k = 1;
    if (Z.size() && Z.front() > x)
        x = Z.front(), k = 2;
    if (k == 0)
        X.pop();
    else if (k == 1)
        Y.pop();
    else
        Z.pop();
    return x;
}

int main()
{
    int n, m, q, u, v, t;
    scanf("%d%d%d%d%d%d", &n, &m, &q, &u, &v, &t);
    double p = (double)u / v;
    for (int i = 1; i <= n; ++i)
        scanf("%d", A + i);
    sort(A + 1, A + n + 1);
    for (int i = n; i; --i)
        X.push(A[i]);
    int d = 0;
    for (int i = 1; i <= m; ++i)
    {
        int x = get() + d;
        cut[i] = x;
        d += q;
        int y = x * p, z = x - y;
        Y.push(y - d);
        Z.push(z - d);
    }
    for (int i = 1; i <= n + m; ++i)
        rem[i] = get() + d;
    for (int i = t; i <= m; i += t)
        printf("%d ", cut[i]);
    puts("");
    for (int i = t; i <= m + n; i += t)
        printf("%d ", rem[i]);
    puts("");
    return 0;
}