Arrangement for Contests UVALive - 8519

最新推荐文章于 2022-03-16 21:36:19 发布

原创最新推荐文章于 2022-03-16 21:36:19 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#贪心 #线段树

思维同时被 2 个专栏收录

147 篇文章

订阅专栏

线段树/树状数组/RMQ

124 篇文章

订阅专栏

本文介绍了一种利用线段树进行贪心操作的算法实现方法。通过维护区间最小值并更新，解决了一系列动态变化的问题。该算法适用于需要快速查询区间最小值并删除的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点击打开链接

贪心的思想线段树操作

每次取区间最小值然后拿掉这个最小值并不影响区间内非最小值的匹配

#include <bits/stdc++.h>
using namespace std;
#define N 0x3f3f3f3f

struct node1
{
    int l;
    int r;
    int minn;
    int pos;
    int laz;
};

struct node2
{
    int minn;
    int pos;
};

node1 tree[400010];
int n,p;

void pushup(int cur)
{
    if(tree[2*cur].minn<tree[2*cur+1].minn)
    {
        tree[cur].minn=tree[2*cur].minn;
        tree[cur].pos=tree[2*cur].pos;
    }
    else
    {
        tree[cur].minn=tree[2*cur+1].minn;
        tree[cur].pos=tree[2*cur+1].pos;
    }
    return;
}

void pushdown(int cur)
{
    if(tree[cur].laz!=0)
    {
        tree[2*cur].minn-=tree[cur].laz;
        tree[2*cur].laz+=tree[cur].laz;
        tree[2*cur+1].minn-=tree[cur].laz;
        tree[2*cur+1].laz+=tree[cur].laz;
        tree[cur].laz=0;
    }
    return;
}

void build(int l,int r,int cur)
{
    int m;
    tree[cur].l=l;
    tree[cur].r=r;
    tree[cur].minn=N;
    tree[cur].pos=-1;
    tree[cur].laz=0;
    if(l==r)
    {
        scanf("%d",&tree[cur].minn);
        tree[cur].pos=l;
        return;
    }
    m=(l+r)/2;
    build(l,m,2*cur);
    build(m+1,r,2*cur+1);
    pushup(cur);
    return;
}

node2 query(int pl,int pr,int cur)
{
    node2 res,tem;
    if(pl<=tree[cur].l&&tree[cur].r<=pr)
    {
        res.minn=tree[cur].minn;
        res.pos=tree[cur].pos;
        return res;
    }
    res.minn=N,res.pos=-1;
    if(pl<=tree[2*cur].r)
    {
        tem=query(pl,pr,2*cur);
        if(res.minn>tem.minn) res=tem;
    }
    if(pr>=tree[2*cur+1].l)
    {
        tem=query(pl,pr,2*cur+1);
        if(res.minn>tem.minn) res=tem;
    }
    return res;
}

void update(int pl,int pr,int val,int cur)
{
    if(pl<=tree[cur].l&&tree[cur].r<=pr)
    {
        tree[cur].minn-=val;
        tree[cur].laz+=val;
        return;
    }
    pushdown(cur);
    if(pl<=tree[2*cur].r) update(pl,pr,val,2*cur);
    if(pr>=tree[2*cur+1].l) update(pl,pr,val,2*cur+1);
    pushup(cur);
    return;
}

int main()
{
    node2 tem;
    int t,k,ans;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&k);
        build(1,n,1);
        ans=0;
        p=1;
        while(p+k-1<=n)
        {
            tem=query(p,p+k-1,1);
            ans+=tem.minn;
            update(p,p+k-1,tem.minn,1);
            p=tem.pos+1;
        }
        printf("%d\n",ans);
    }
    return 0;
}