专题五算法优化——优先队列

题目链接

题目如下：

题目大意：

解题思路：

本题不难，但是要注意细节。

当i大于m时，只要保证即可

当i小于等于m时，只要保证即可

因为要求的是最小的操作次数，而如果到循环到k时不满足，就应该找出到中最大(小)的数，而如果每次都重新查询一次会TLE，那就使用优先队列！

优先队列介绍

其中的细节：队列每次都要变或者开多个队列（wa了一晚上）

代码如下：

#include <cstdio>
#include <queue>
#include <deque>
#include <algorithm>
#include <iostream>

inline void read(long long &x) {
    x = 0;
    short flag = 1;
    char c = getchar();
    while (c < '0' || c > '9') {
        if (c == '-')
            flag = -1;
        c = getchar();
    }
    while (c >= '0' && c <= '9') {
        x = (x << 3) + (x << 1) + (c ^ 48);
        c = getchar();
    }
    x *= flag;
}
long long a[212345];
long long sum[212345];
long long resum[212345];

int main() {
    long long T;
    read(T);
    while (T--) {
        std::priority_queue<long long, std::deque<long long>, std::greater<int>> q1;
        std::priority_queue<long long> q2;
        long long n, m;
        long long ans = 0;
        read(n);
        read(m);

        for (int i = 1; i <= n; i++) {
            read(a[i]);

            if (i > m) {

                q1.push(a[i]);
                if (i == m + 1)
                    sum[i] = a[i];
                else
                    sum[i] = sum[i - 1] + a[i];

                if ( sum[i] < 0) {
                    sum[i] -= 2 * q1.top();
                    q1.pop();
                    ans++;
                }
            }

        }
        for (int i = m; i >= 2; i--) {
            q2.push(a[i]);
            if (i == m)
                resum[i] = a[i];
            else
                resum[i] = resum[i + 1] + a[i];
            if (resum[i] > 0) {
                resum[i] -=  q2.top() * 2;
                q2.pop();
                ans++;
            }
        }

        printf("%d\n", ans);
    }
}