HDU 1754:I Hate It(线段树模板)

I Hate It

Time Limit: 9000/3000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 102748    Accepted Submission(s): 38640

Problem Description

很多学校流行一种比较的习惯。老师们很喜欢询问,从某某到某某当中,分数最高的是多少。
这让很多学生很反感。

不管你喜不喜欢,现在需要你做的是,就是按照老师的要求,写一个程序,模拟老师的询问。当然,老师有时候需要更新某位同学的成绩。

Input

本题目包含多组测试,请处理到文件结束。
在每个测试的第一行,有两个正整数 N 和 M ( 0<N<=200000,0<M<5000 ),分别代表学生的数目和操作的数目。
学生ID编号分别从1编到N。
第二行包含N个整数,代表这N个学生的初始成绩,其中第i个数代表ID为i的学生的成绩。
接下来有M行。每一行有一个字符 C (只取'Q'或'U') ,和两个正整数A,B。
当C为'Q'的时候,表示这是一条询问操作,它询问ID从A到B(包括A,B)的学生当中,成绩最高的是多少。
当C为'U'的时候,表示这是一条更新操作,要求把ID为A的学生的成绩更改为B。

Output

对于每一次询问操作,在一行里面输出最高成绩。

Sample Input

5 6
1 2 3 4 5
Q 1 5
U 3 6
Q 3 4
Q 4 5
U 2 9
Q 1 5

Sample Output

5
6
5
9

Hint

Huge input,the C function scanf() will work better than cin

思路

线段树的单点更新模板题,用一个结构体记录区间的端点值,以及每个区间对应的权值。

build函数实现了对一个区间建造以i为祖先的线段树

update用来更新线段树

ququery用来查询区间最大值

详细的讲解在代码注释里

AC代码

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <string>
#define ll long long
#define ull unsigned long long
#define ms(a) memset(a,0,sizeof(a))
#define pi acos(-1.0)
#define INF 0x7f7f7f7f
const double E=exp(1);
const int maxn=2e5+10;
const int mod=1e9+7;
using namespace std;
int MAX;
struct wzy
{
	int left,right;
	int value;//对应区间的最大值
}p[maxn<<2];
int father[maxn];//记录了每个点对应结构体数组的下标
// 为区间[l,r]建立一个以i为祖先的线段树
void build(int i,int l,int r)
{
	p[i].left=l;//写入第i个节点的左区间
	p[i].right=r;//写入右区间
	p[i].value=0;//每个区间的最大值初始为0
	if(l==r)//当区间长度为0时,停止递归
	{
		father[l]=i;
		return ;
	}
	// 该节点往左孩子的方向继续建立线段树
	build(i<<1,l,(l+r)/2);
	// 往右孩子方向建立线段树
	build((i<<1)+1,(l+r)/2+1,r);
}
// 单点更新线段树
// 从下往上更新(该点已经在函数外更新过了)
void update(int ri)
{
	// 已经找到了祖先(整个线段树的祖先节点对应的下标为1)
	if(ri==1)
		return ;
	// fi为ri的父亲节点
	int fi=ri/2;
	// 父亲节点的两个孩子
	int l=p[fi<<1].value;
	int r=p[(fi<<1)+1].value;
	// 更新父亲节点的值
	p[fi].value=max(l,r);
	// 递归更新,从父亲节点往上找
	update(ri/2);
}
// 查询区间最大值
// 将一段区间按照建立的线段树从上往下一直拆开,直到找到和需要查询的区间完全重合停止
// i为区间的序号,一般是1
void query(int i,int l,int r)
{	
	// 找到了一个完全重合的区间
	if(p[i].left==l&&p[i].right==r)
	{
		// 更新最大值
		MAX=max(MAX,p[i].value);
		return ;
	}
	// 左儿子
	i=i<<1;
	// 如果左区间有涉及
	if(l<=p[i].right)
	{
		// 全包含在左区间内
		if(r<=p[i].right)
			query(i,l,r);
		// 有部分包含在左区间内,查询区间拆分,左端点不变,右端点变成做孩子的右区间端点
		else
			query(i,l,p[i].right);
	}
	// 右儿子
	i+=1;
	// 和左区间的操作一样
	if(r>=p[i].left)
	{
		if(l>=p[i].left)
			query(i,l,r);
		else
			query(i,p[i].left,r);
	}
}	
int main(int argc, char const *argv[])
{
	int n,m;
	int x;
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		// 初始化建图
		build(1,1,n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			p[father[i]].value=x;
			// 更新线段树
			update(father[i]);
		}
		char ch[3];
		int a,b;
		while(m--)
		{
			MAX=INT_MIN;
			scanf("%s%d%d",ch,&a,&b);
			if(ch[0]=='Q')
			{
				query(1,a,b);//查询a,b区间的最大值
				cout<<MAX<<endl;
			}
			else
			{
				p[father[a]].value=b;//更新下标为father[a]的点的值
				update(father[a]);
			}
		}
	}
	return 0;
}

上面的那种写法有点慢,因为在输入前建了一次图,输入时又建了一次图,操作重复了。下面是只建一次图的写法

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <limits.h>
#include <map>
#include <stack>
#include <queue>
#include <vector>
#include <set>
#include <string>
#define ll long long
#define ull unsigned long long
#define ms(a) memset(a,0,sizeof(a))
#define pi acos(-1.0)
#define INF 0x7f7f7f7f
const double E=exp(1);
const int maxn=2e5+10;
const int mod=1e9+7;
using namespace std;
#define ls o<<1
#define rs o<<1|1
int a[maxn];
struct node
{
	int l,r;
	int x;
}t[maxn<<2];
inline void push_up(int o) 
{
    t[o].x = max(t[ls].x,t[rs].x);
}
void build(int l,int r,int o) 
{
    t[o].l=l, t[o].r=r;
    if(l==r)  
	{
		t[o].x=a[l]; 
		return;
	}
    int mid=(l+r)>>1;
    build(l,mid,ls); 
    build(mid+1,r,rs);
    push_up(o);
}
inline void update(int s, int o, int x) 
{
    if(t[o].l==t[o].r && t[o].l==s) 
    {
        t[o].x=x;
        return;
    }
    int mid=(t[o].l+t[o].r)>>1;
    if(s<=mid) 
    	update(s,ls,x);
    else 
    	update(s,rs,x);
    push_up(o);
}
inline int query(int l,int r, int o) 
{
    if(t[o].l>=l&&t[o].r<=r) 
    {
    	return t[o].x;
    }
    int mid=(t[o].l+t[o].r)>>1;
    if(r<=mid) 
    	return query(l,r,ls);
    else if(l>mid) 
    	return query(l,r,rs);
    else 
    {
        return max(query(l,mid,ls),query(mid+1,r,rs));
    }
}
int main(int argc, char const *argv[])
{
	int n,q;
	char ch[1];
	while(scanf("%d%d",&n,&q)!=EOF)
	{
		for(int i=1;i<=n;i++)
			scanf("%d",&a[i]);
		build(1,n,1);
		while(q--)
		{
			int L,R,X;
			scanf("%s",ch);
			if(ch[0]=='Q')
			{
				scanf("%d%d",&L,&R);
				cout<<query(L,R,1)<<endl;
			}
			else
			{
				scanf("%d%d",&L,&X);
				update(L,1,X);
			}
		}
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值