在上一篇博文里面,我提到了我不会线段树,现在就努力地学习啊!
今天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; }