P1792 贪心 + 优先队列

最新推荐文章于 2021-10-20 16:38:28 发布

SHOHOKUKU

最新推荐文章于 2021-10-20 16:38:28 发布

阅读量96

点赞数

分类专栏：基本算法

本文链接：https://blog.csdn.net/neweryyy/article/details/109466752

版权

基本算法专栏收录该内容

252 篇文章 1 订阅

订阅专栏

题意

传送门 P1792 [国家集训队]种树

题解

P1484 种树的环版本。贪若心地取权值最大的位置种树，可能出现取相邻位置种树更优的情况。设位置 $i$ 的权值为 $P_i$ ，那么每次贪心选择后，若 $P_{pre[i]}+P_{nxt[i]}>P_i$ ，应该取消位置 $i$ 的选择，而选择相邻位置，此时保证总选择位置数量增加 $1$ 。那么每次将 $i$ 处权值赋为 $P_{pre[i]}+P_{nxt[i]}-P_i$ 后加入优先队列，并删除相邻节点。

#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
#define maxn 200005
struct node
{
    int a, id;
    bool operator<(const node &b) const
    {
        return a < b.a;
    }
};
int N, M, A[maxn];
int pre[maxn], nxt[maxn], del[maxn];
priority_queue<node> q;

int main()
{
    scanf("%d%d", &N, &M);
    for (int i = 1; i <= N; ++i)
    {
        scanf("%d", A + i);
        pre[i] = i - 1, nxt[i] = i + 1;
        q.push(node{A[i], i});
    }
    pre[1] = N, nxt[N] = 1;
    if (M > N / 2)
    {
        puts("Error!");
        return 0;
    }
    int res = 0;
    for (int i = 0; i < M; ++i)
    {
        while (del[q.top().id])
            q.pop();
        node t = q.top();
        q.pop();
        res += t.a;
        A[t.id] = A[pre[t.id]] + A[nxt[t.id]] - A[t.id];
        del[pre[t.id]] = del[nxt[t.id]] = 1;
        pre[t.id] = pre[pre[t.id]], nxt[t.id] = nxt[nxt[t.id]];
        nxt[pre[t.id]] = t.id, pre[nxt[t.id]] = t.id;
        q.push(node{A[t.id], t.id});
    }
    printf("%d\n", res);
    return 0;
}