HDU 1754 I Hate It
传送门
中文题,题意自己看。
线段树入门之入门题
但是贡献了若干RE,据说是要开到60w才不会爆
具体没有试
下面讲讲线段树
基本思路和树状数组一样,通过增加更新时间,来减少查询时间。
建一个二叉树
每个点表示一个管辖范围
如1-7
树如下
不难发现左子树的线段为 [l,mid] 右子树为[mid+1,r]
这样就先建树吧
void BuildTree(int num,int l,int r)
{
Tree[num].l=l;
Tree[num].r=r;
if(r==l)
{
Tree[num].v=a[l];//a[l]为最早的叶子节点
return ;
}
int mid=(l+r)>>1;
BuildTree(num<<1,l,mid);
BuildTree(num<<1|1,mid+1,r);
Tree[num].v=max(Tree[num<<1].v,Tree[num<<1|1].v);
//更新父节点
}
下面考虑如何更新树
先找到叶子节点,再向上更新
可以再建树时再开个数组记录叶子节点位置
可以减少一点点复杂度
if(r==l)
{
Tree[num].v=a[l];//a[l]为最早的叶子节点
w[l]=num;//记录叶子节点位置
return ;
}
这里直接找了节点,慢不了多少,都差不多。
更新就是观察它和它的左子树或右子树哪个值大,就把值给父亲节点
void Update(int num,int a,int b)
{
if(Tree[num].l==a&&Tree[num].r==a)
{
Tree[num].v=b;
return ;
}
int mid=(Tree[num].l+Tree[num].r)>>1;
if(a<=mid)Update(num<<1,a,b);
if(a>mid)Update(num<<1|1,a,b);
Tree[num].v=max(Tree[num<<1].v,Tree[num<<1|1].v);
}
最后就是查询了
之前写了一个查询,效率不高,看了大神的代码
学习了一下
int Querry(int num,int l,int r)
{
if(Tree[num].l==l&&Tree[num].r==r)
return Tree[num].v;
int mid=(Tree[num].l+Tree[num].r)>>1;
if(r<=mid)return Querry(num<<1,l,r);
if(l>mid)return Querry(num<<1|1,l,r);
if(l<=mid&&r>mid)
return max(Querry(num<<1,l,mid),Querry(num<<1|1,mid+1,r));
return 0;
}
之前自己写的是先找到[l,n]
再从n+1重新开始找
直到找到 [m,r]
这里有个问题就是每次又从头开始找
浪费了不少的时间
大神则巧妙的用递归解决了这个问题 Orz
这道题还有个坑就是读入
用%s读可以避开这个问题
同时此题数据巨水
错误代码都能AC……………….
可以加输入输出外挂
优化刚刚的
具体代码如下
#include<cstdio>
const int MAXN=200000;
int n,m,a[MAXN+5],x,y;
char c;
struct Tree
{
int l,r,v;
}Tree[MAXN<<2];
inline int read()
{
char c=getchar();
if(c==EOF) return 0;
while(!(c>='0'&&c<='9'))c=getchar();
int x=0;
while(c>='0'&&c<='9')
x=x*10+c-'0',c=getchar();
return x;
}
int buf[10];
inline void put(int i)
{
int p=0;
if(i==0)p++;
else while(i)
{
buf[p++]=i%10;
i/=10;
}
for(int j=p-1;j>=0;j--)putchar('0'+buf[j]);
putchar('\n');
}
inline int max(int a,int b)
{
return a>b?a:b;
}
void BuildTree(int num,int l,int r)
{
Tree[num].l=l;
Tree[num].r=r;
if(r==l)
{
Tree[num].v=a[l];
return ;
}
int mid=(l+r)>>1;
BuildTree(num<<1,l,mid);
BuildTree(num<<1|1,mid+1,r);
Tree[num].v=max(Tree[num<<1].v,Tree[num<<1|1].v);
}
void Update(int num,int a,int b)
{
if(Tree[num].l==a&&Tree[num].r==a)
{
Tree[num].v=b;
return ;
}
int mid=(Tree[num].l+Tree[num].r)>>1;
if(a<=mid)Update(num<<1,a,b);
if(a>mid)Update(num<<1|1,a,b);
Tree[num].v=max(Tree[num<<1].v,Tree[num<<1|1].v);
}
int Querry(int num,int l,int r)
{
if(Tree[num].l==l&&Tree[num].r==r)
return Tree[num].v;
int mid=(Tree[num].l+Tree[num].r)>>1;
if(r<=mid)return Querry(num<<1,l,r);
if(l>mid)return Querry(num<<1|1,l,r);
if(l<=mid&&r>mid)
return max(Querry(num<<1,l,mid),Querry(num<<1|1,mid+1,r));
return 0;
}
int main()
{
while(n=read())
{
m=read();
for(int i=1;i<=n;i++)
a[i]=read();
BuildTree(1,1,n);
for(int i=1;i<=m;i++)
{
c=getchar();
x=read();
y=read();
if(c=='Q')
put(Querry(1,x,y));
else
Update(1,x,y);
}
}
return 0;
}
附上一些自己的测试数据
2 7
3 3
U 1 5
U 2 4
U 2 4
U 1 9
U 2 2
Q 1 2
Q 2 2
20 17
5 2 7 1 6 6 8 1 2 9 8 1 5 6 7 4 2 7 8 4
U 16 7
U 19 8
Q 11 17
U 13 2
U 4 1
Q 3 11
Q 5 9
U 8 2
U 7 7
U 14 8
U 6 8
Q 1 10
Q 3 16
Q 6 9
Q 7 9
U 6 7
U 7 5
4 11
3 6 5 8
Q 1 4
Q 3 4
U 3 9
U 1 9
Q 4 4
U 2 8
U 1 5
Q 2 3
Q 1 1
Q 1 3
Q 1 3
8 15
2 8 8 9 3 1 3 3
Q 2 2
Q 5 8
U 5 4
U 1 2
Q 1 8
U 8 9
U 4 9
U 7 6
Q 2 5
Q 1 7
U 6 7
Q 1 2
U 2 8
U 7 2
U 4 3
1 2
8
U 1 3
Q 1 1
15 9
6 4 8 9 3 3 8 2 2 4 4 8 5 1 3
Q 2 3
U 6 8
U 11 7
Q 5 8
U 10 3
U 11 3
U 1 7
U 14 8
U 14 4
17 16
5 7 9 6 9 3 4 5 3 3 3 4 4 6 1 2 8
Q 10 16
U 2 6
U 6 1
Q 9 15
Q 7 17
Q 14 17
U 1 7
Q 2 14
Q 11 12
Q 11 16
U 17 3
U 1 9
Q 4 4
Q 5 14
Q 7 16
Q 4 12
2 3
8 9
U 1 7
U 1 2
Q 2 2
2 6
1 5
Q 2 2
Q 2 2
Q 1 1
Q 2 2
U 2 8
U 2 6
20 2
2 3 8 3 5 2 2 2 1 5 4 8 5 8 5 8 5 7 2 9
Q 1 12
U 15 6
ans
9
2
8
9
8
9
9
8
7
8
8
8
9
5
9
9
8
3
9
9
9
8
3
8
8
6
6
8
8
9
4
6
6
9
6
9
9
5
5
1
5
8