蓝桥杯模拟赛最后一题 晚会节目单(线段树)

明天就省赛了，参加的人少，选出10%作为省一感觉有点难度。希望我能再次得个省一，再去次北京，最后能有个好结果吧。大三就结束了，Time flies.
【问题描述】
小明要组织一台晚会，总共准备了 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<iostream>
using namespace std;
int n,m;
int data[100005];
struct point{
	int l,r;
	int pom;//pos of max
};
point tree[400005];
void pushup(int k){
	tree[k].pom=data[tree[2*k].pom]>data[tree[2*k+1].pom]?tree[2*k].pom:tree[2*k+1].pom;
}
void build(int k,int a,int b){//树的节点编号，区间左右边界 
	tree[k].l=a;
	tree[k].r=b;
	if(a==b){
		tree[k].pom=a;
		return ;
	}
	int m=(a+b)/2;
	build(2*k,a,m);
	build(2*k+1,m+1,b);
	pushup(k);
}
int query(int k,int a,int b){
	//在闭区间(a,b)内查找
	if(tree[k].l==a&&tree[k].r==b){
		return tree[k].pom;
	} 
	int m=(tree[k].l+tree[k].r)/2;
	if(b<=m){
		return query(2*k,a,b);
	}else if(a>m){
		return query(2*k+1,a,b);
	}else{
		int ans1=query(2*k,a,m);
		int ans2=query(2*k+1,m+1,b);
		return data[ans1]>data[ans2]?ans1:ans2;
	}
} 
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;++i){
		cin>>data[i];
	}
	build(1,1,n);
	int pos1=1,pos2=n-m+1;//尺取法，在一个区间内求最大值
	int pos_m=0;
	while(pos1<pos2&&pos2<=n){
		//找到(pos1,pos2)闭区间的最大值下标 
		pos_m=query(1,pos1,pos2);
		cout<<data[pos_m]<<" ";
		pos1=pos_m+1;
		pos2++;
	}
	while(pos2<=n){
		cout<<data[pos2++]<<" ";
	}
	cout<<endl;
	return 0;
}