在上一篇博文里面,我提到了我不会线段树,现在就努力地学习啊!
今天AC一题感觉都很累,可能是状态不佳,在做HDU1166这题目时候,RE了无数次。
原因是:我的宏定义写错了,我已经不是第一犯这种错误了!我的宏写成了
#ifndef ONLINE_JUDEGE
freopen("in.txt","r",stdin);
#endif
就是这个错误我调了两个小时……教训!!!每次出现RE我都觉得是我的数组越界,但是原来RE也有这种情况。
后来我又TLE了!原因是,C++读写太慢了,并且我做了不必要的比较。
之后我WA了两次。
原因:我的Case i:没有增加……以及每次都输出前都赋值为1了!没有写在外面!!!
低级错误啊都是……
回归正题:
线段树,目前我理解的线段树就是一颗满二叉树,依赖着满二叉的二分性质,成段更新,以提升速度。
在网上看了不少人的文章,大多是用函数来写的,我还是习惯封装成为类,自我感觉更易懂。
现在初步在学习的是单点线段树,就像HDU1166
来看看HDU1166,这题目很搞啦。
1 10 1 2 3 4 5 6 7 8 9 10 Query 1 3 Add 3 6 Query 2 7 Sub 10 2 Add 6 3 Query 3 10 End
Case 1: 6 33 59
做法就是线段树咯。无非一个是插入,一个是查询,其过程更二叉树很像,但是有上滤计算。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<string>
using namespace std;
#define lc l,m,root<<1
#define rc m+1,r,root<<1|1
#define mid int m =(l+r)>>1
class Tree
{
public:
Tree() {};
Tree(int n)
{
this->n=n;
cover.resize(n<<2);
Build(1,n,1);
}
~Tree()
{
cover.clear();
}
void Insert(int p,int cov)
{
Insert(p,cov,1,n,1);
}
int Find(int l,int r)
{
return Find(l,r,1,n,1);
}
void show()
{
show(1,n,1);
}
private:
vector<int> cover;
int n;
void Build(int l,int r,int root)
{
if(l==r)
{
scanf("%d",&cover[root]);
return;
}
mid;
Build(lc);
Build(rc);
PushUp(root);
}
void PushUp(int root)
{
cover[root]=cover[root<<1]+cover[root<<1|1];
}
void Insert(int p,int cov,int l,int r,int root)
{
if(l==r)
{
cover[root]+=cov;
return ;
}
mid;
if(p<=m) Insert(p,cov,lc);//左儿子2*n
else Insert(p,cov,rc);//右儿子2*n+1
PushUp(root);
}
int Find(int L,int R,int l,int r,int root)
{
if(L<=l&&R>=r)
{
return cover[root] ;
}
int sum=0;
mid;
if(R>m)
{
sum+= Find(L,R,rc);//右儿子
}
if(L<=m)
{
sum+= Find(L,R,lc);//左儿子
}
return sum;
}
void show(int l,int r,int root)
{
if(l>=r) return;
mid;
show(lc );
printf("%d\n",cover[root]);
show(rc);
}
};
int main()
{
int L,R,N;
int ind;
int cases;
char str[10];
scanf("%d",&cases);
ind=1;
while(cases--)
{
scanf("%d",&N);
Tree t(N);
printf("Case %d:\n",ind++);
while(scanf("%s",str))
{
if(str[0]==‘E‘) break;
scanf("%d%d",&L,&R);
if(str[0]==‘Q‘)
{
printf("%d\n",t.Find(L,R));
}
else if(str[0]==‘A‘)
{
t.Insert(L,R);
}
else if(str[0]==‘S‘)
{
t.Insert(L,-R);
}
}
}
return 0;
}
1742基本一样啊!就是PUSHUP不一样。
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
基本上我的改动就是取PUSHUP和FIND的时候取的是MAX。一次AC了 T_T,感动啊!!!5 6 5 9
#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<string>
#include<cmath>
using namespace std;
#define lc l,m,root<<1
#define rc m+1,r,root<<1|1
#define mid int m =(l+r)>>1
class Tree
{
public:
Tree() {};
Tree(int n)
{
this->n=n;
cover.resize(n<<2);
Build(1,n,1);
}
~Tree()
{
cover.clear();
}
void Insert(int p,int cov)
{
Insert(p,cov,1,n,1);
}
int Find(int l,int r)
{
return Find(l,r,1,n,1);
}
void show()
{
show(1,n,1);
}
private:
vector<int> cover;
int n;
void Build(int l,int r,int root)
{
if(l==r)
{
scanf("%d",&cover[root]);
return;
}
mid;
Build(lc);
Build(rc);
PushUp(root);
}
void PushUp(int root)
{
cover[root]=max(cover[root<<1],cover[root<<1|1]);
}
void Insert(int p,int cov,int l,int r,int root)
{
if(l==r)
{
cover[root]=cov;
return ;
}
mid;
if(p<=m) Insert(p,cov,lc);//左儿子2*n
else Insert(p,cov,rc);//右儿子2*n+1
PushUp(root);
}
int Find(int L,int R,int l,int r,int root)
{
if(L<=l&&R>=r)
{
return cover[root] ;
}
int sum=0;
mid;
if(R>m)
{
sum=max(sum, Find(L,R,rc) );//右儿子
}
if(L<=m)
{
sum=max(sum, Find(L,R,lc));//左儿子
}
return sum;
}
void show(int l,int r,int root)
{
if(l>=r) return;
mid;
show(lc );
printf("%d\n",cover[root]);
show(rc);
}
};
int main()
{
int L,R,N,M;
char str[10];
while( scanf("%d%d",&N,&M)!=EOF){
Tree t(N);
// t.show();
while(M--)
{
scanf(" %c ",&str[0]);
scanf("%d%d",&L,&R);
if(str[0]==‘Q‘)
{
printf("%d\n",t.Find(L,R));
}
else
{
t.Insert(L,R);
}
}
}
return 0;
}
线段树(单点更新)HDU1166、HDU1742,布布扣,bubuko.com
原文:http://blog.csdn.net/dengyaolongacmblog/article/details/23361489