hihoCoder1299 打折机票(线段树)

题目来源

http://hihocoder.com/problemset/problem/1299

描述

因为思念新宿的"小姐姐"们，岛娘计划6月份再去一趟东京，不过这次看来她需要自掏腰包。经过了几天的夜战，岛娘终于在体力耗尽之前，用Python抓下了所有6月份，上海至东京的全部共 n 张机票。现在请你帮助债台高筑的岛娘筛选出符合时间区间要求的，最贵的机票。
输入
输入数据的第一行包含两个整数 n, m(1 ≤ n, m ≤ 1e5)，分别表示机票的总数，和询问的总数。接下来的 n 行，每行两个整数 t, v (1 ≤ t, v ≤ 1e5)，表示每张机票出发的时间和价格。 接下来的 m 行，每行两个整数 a, b (1 ≤ a ≤ b ≤ 1e5)，表示每个询问所要求的时间区间。
输出
对于每组询问，输出一行表示最贵的价格。如果没有符合要求的机票，输出一行"None"。
样例输入
7 6
1 1
2 1
4 3
4 4
4 5
6 9
7 9
1 7
1 2
6 7
3 3
4 4
5 5
样例输出
9
1
9
None
5

None

分析

试了各种方法（set、map存储数据，二分查找）都是超时。。。超时应该是在下标a~b查找最大值的过程，所以应该采用线段树。

代码

#include <iostream>
using namespace std;
#include <string>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#define EACH(it,a) for(auto it=begin(a);it!=end(a);it++)
#define LL long long
#define INF 0x3f3f3f3f
const int SZ=100002;

int pri[SZ];

struct Tree{
	int l;
	int r;
	int mx;
}tree[4*SZ]; //用数组存储树, hihocoder题目大小为2*SZ时运行出错

void build(int i,int l,int r) //注意要有参数i
{
	tree[i].l=l;
	tree[i].r=r;
	if(l==r)
		tree[i].mx=pri[l];
	else{
		int mid=(l+r)/2;
		build(2*i,l,mid);
		build(2*i+1,mid+1,r);
		tree[i].mx=max(tree[2*i].mx,tree[2*i+1].mx);
	}
}

int query(int i,int l,int r) //三种情况
{
	if(tree[i].l>r || tree[i].r<l)
		return 0;
	if(tree[i].l>=l && tree[i].r<=r)
		return tree[i].mx;
	else
		return max(query(2*i,l,r),query(2*i+1,l,r));
}

int main()
{
	ios_base::sync_with_stdio(0);
#ifdef LJY
	freopen("in.txt","r",stdin);
#endif
	int i,j;
	int n,m;
	int a,b;
	int ans;
	cin>>n>>m;
	int t,v;
	int mi=SZ,mx=0;
	for(i=0;i<n;i++){
		cin>>t>>v;
		pri[t]=max(pri[t],v);
		mi=min(mi,t);
		mx=max(mx,t);
	}

	build(1,mi,mx); //直接用1-1e5也行

	while(m--){
		cin>>a>>b;
		a=max(a,mi);
		b=min(b,mx);
		ans=0;
		
		ans=query(1,a,b);
		if(ans!=0)
			cout<<ans<<endl;
		else
			cout<<"None"<<endl;
	}

	return 0;
}

其中query函数 

return max(query(2*i,l,r),query(2*i+1,l,r));

不要去采用

return max(query(2*i,l,mid),query(2*i+1,mid+1,r));