其他分享
首页 > 其他分享> > 第十一届蓝桥杯校内赛:选节目(线段树解法)

第十一届蓝桥杯校内赛:选节目(线段树解法)

作者:互联网

选节目

小明要组织一台晚会,总共准备了 n 个节目。然后晚会的时间有限,他只能最终选择其中的 m 个节目。这 n 个节目是按照小明设想的顺序给定的,顺序不能改变。
小明发现,观众对于晚上的喜欢程度与前几个节目的好看程度有非常大的关系,他希望选出的第一个节目尽可能好看,在此前提下希望第二个节目尽可能好看,依次类推。
小明给每个节目定义了一个好看值,请你帮助小明选择出 m 个节目,满足他的要求。

输入格式

输入的第一行包含两个整数 n, m ,表示节目的数量和要选择的数量。
第二行包含 n 个整数,依次为每个节目的好看值。

输出格式

输出一行包含 m 个整数,为选出的节目的好看值。

样例输入

5 3
3 1 2 5 4

样例输出

3 5 4

样例说明

选择了第1, 4, 5个节目。
评测用例规模与约定
对于 30% 的评测用例,1 <= n <= 20;
对于 60% 的评测用例,1 <= n <= 100;
对于所有评测用例,1 <= n <= 100000,0 <= 节目的好看值 <= 100000。

线段树区间返回区间最值及下标
#include <bits/stdc++.h>
using namespace std;
static const int maxn = 1e5 + 5;
static const int mod = 10000;
typedef long long ll;
int a[maxn];
struct node
{
    int id;
    int left,right;
    int max;
}tr[maxn << 2];

pair<int,int> buildmax(int left,int right,int now)
{
    tr[now].left=left;
    tr[now].right=right;
    if(left==right)
    {
        tr[now].id=left;
        tr[now].max=a[left];
        return make_pair(left,a[left]);
    }
    int mid=(left+right)>>1;

    pair<int,int> r1=buildmax(left,mid,now*2);
    pair<int,int> r2=buildmax(mid+1,right,now*2+1);
    if(r1.second>=r2.second)
    {
        tr[now].id=r1.first;
        tr[now].max=r1.second;
        return r1;
    }
    else
    {
        tr[now].id=r2.first;
        tr[now].max=r2.second;
        return r2;
    }
}

pair<int,int> querymax(int left,int right,int now)
{
    if(left == tr[now].left && right == tr[now].right)
        return make_pair(tr[now].id,tr[now].max);
    int mid=(tr[now].left+tr[now].right)>>1;

    if(right<=mid)
        return querymax(left, right, now << 1);
    else if(left>mid)
        return querymax(left, right, now << 1 | 1);
    else
    {
        pair<int,int> r1=querymax(left, mid, now << 1);
        pair<int,int> r2=querymax(mid+1, right, now << 1 | 1);
        return r1.second >= r2.second ? r1 : r2;
    }
}
int main(){
    int n, m;
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
    buildmax(1, n, 1);
    int ql = 1, qr = n - m + 1;
    for(int i = 1; i <= m; ++i){
        pair<int, int>res = querymax(ql, qr, 1);
        printf("%d%c", res.second, i == m ? '\n' : ' ');
        ql = res.first + 1;
        qr++;
    }
    return 0;
}

标签:right,校内,int,线段,tr,蓝桥,now,节目,left
来源: https://blog.csdn.net/qq1013459920/article/details/104881742