2017 ICPC Asia Xian Onsite H. Arrangement for Contests #线段树#

2020-01-29 13:07:09 阅读：292 来源： 互联网

标签：Onsite ch return int tree Xian mid Contests id

Portal

很容易想到贪心+线段树查询维护

UVa上这题是假数据 O(nk)暴力都能0ms AC...

此外输入可能是空文件坑！！！

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

typedef long long ll;
const int maxn = 1e5 + 10;
struct node { int l, r, minv, minp, lz; } tree[maxn << 2];
struct Min
{
    int v, p;
    Min(int v, int p) { this->v = v; this->p = p; }
    bool operator < (const Min& x) const { return v == x.v ? p > x.p : v < x.v;  }
};

inline const int read()
{
    int x = 0, f = 1; char ch = getchar();
    while (ch < '0' || ch > '9') { if (ch == '-') f = -1; ch = getchar(); }
    while (ch >= '0' && ch <= '9') { x = (x << 3) + (x << 1) + ch - '0'; ch = getchar(); }
    return x * f;
}

inline int ls(int id) { return id << 1; }

inline int rs(int id) { return id << 1 | 1; }

void push_up(int id)
{
    if (tree[rs(id)].minv <= tree[ls(id)].minv)
    {
        tree[id].minv = tree[rs(id)].minv;
        tree[id].minp = tree[rs(id)].minp;
    }
    else
    {
        tree[id].minv = tree[ls(id)].minv;
        tree[id].minp = tree[ls(id)].minp;
    }
}

void push_down(int id)
{
    if (tree[id].lz)
    {
        tree[ls(id)].minv -= tree[id].lz;
        tree[ls(id)].lz += tree[id].lz;
        tree[rs(id)].minv -= tree[id].lz;
        tree[rs(id)].lz += tree[id].lz;
        tree[id].lz = 0;
    }
}

void build(int id, int l, int r)
{
    tree[id].l = l;
    tree[id].r = r;
    tree[id].lz = 0;
    tree[id].minv = 0x3f3f3f3f;
    if (l == r)
    {
        tree[id].minv = read();
        tree[id].minp = l;
        return;
    }
    int mid = (l + r) >> 1;
    build(ls(id), l, mid);
    build(rs(id), mid + 1, r);
    push_up(id);
}

Min query(int id, int l, int r)
{
    if (tree[id].l == l && tree[id].r == r) return Min(tree[id].minv, tree[id].minp);
    push_down(id);
    int mid = (tree[id].l + tree[id].r) >> 1;
    if (r <= mid) return query(ls(id), l, r);
    if (l > mid) return query(rs(id), l, r);
    return min(query(ls(id), l, mid), query(rs(id), mid + 1, r));
}

void update(int id, int l, int r, int sub)
{
    if (tree[id].l == l && tree[id].r == r)
    {
        tree[id].minv -= sub;
        tree[id].lz += sub;
        return;
    }
    push_down(id);
    int mid = (tree[id].l + tree[id].r) >> 1;
    if (r <= mid) update(ls(id), l, r, sub);
    else if (l > mid) update(rs(id), l, r, sub);
    else
    {
        update(ls(id), l, mid, sub);
        update(rs(id), mid + 1, r, sub);
    }
    push_up(id);
}

int main()
{
    int t = 0;
    scanf("%d", &t);
    while (t--)
    {
        int n = read(), k = read();
        build(1, 1, n);
        ll res = 0;
        for (int i = 1; i <= n - k + 1; )
        {
            Min m = query(1, i, i + k - 1);
            int minv = m.v, minp = m.p;
            if (minv > 0)
            {
                res += minv;
                update(1, i, i + k - 1, minv);
            }
            i = minp + 1;
        }
        printf("%lld\n", res);
    }
    return 0;
}

SDUWH_2U 发布了367 篇原创文章 · 获赞 148 · 访问量 2万+ 私信关注

标签：Onsite,ch,return,int,tree,Xian,mid,Contests,id
来源： https://blog.csdn.net/qq_35850147/article/details/104105998

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

2017 ICPC Asia Xian Onsite H. Arrangement for Contests #线段树#

Portal