C++算法 线段树
线段树这个算法,看起来非常高端,而且很有用处,所以还是讲一下下吧。
温馨提示:写线段树前请做好写码5分钟,调试一辈子的准备^-^
啊直接步入正题……
首先我们考虑一个题目:有一个序列,要做到单点修改单点查询,该怎么做呢?
同学们先不要着急关掉……我们细细分析,像这种题,明显大家都知道……直接暴力就过了嘛……,所以不做分析……
然后我们考虑第二个题目:有一个序列,要做到单点修改区间求和,该怎么做呢?
像传统的for(int i=1;i<=n;i++)ans+=a[i]当然是很香的,但如果遇到一些非常奇怪的题目(数 据 大 到 你 自 闭),这个就没什么用了……
啊这,简单的暴力不能用,就要用复杂的暴力——线段树,当然是简单亿点点的~因为单点修改可以O(1),不存在浪费时间的事情,所以我们用线段树小小的处理一下就好了。
接下来画一个图帮助斯烤:
在这个图上呢……一共有8个点,就是底下那些最小的线段,而我们把他合成了许多部分,每个部分的值就是f[wz*2]+f[wz*2+1](f是每个的值,wz是每个的编号)。
然后就可以把他一直向下,一直向下,直到发现这个区间是被我们要求的区间所包含的,然后这个区间内的数就全部加上,就这么一直向下找找找……,最后就可以算出一个区间的总和。
下面是演示代码~
void qh(long long wz)//wz是现在所处的小块编号 { if(tree[wz].l>=a&&tree[wz].r<=b)//a和b是我们要求的区间的左右端点 { zshu+=tree[wz].shu;//全部被包含,直接加上 return; } long long mid=(tree[wz].l+tree[wz].r)/2;//不能被包含?分裂一下试试吧 if(b>mid)//和谐小细节~ { qh(wz*2+1);//如果有被包含,就去右边的小块看看 } if(a<=mid)//和谐小细节*2~(至于为啥有和谐小细节,以及为什么要这样写,请同学们自己斯烤(这么简单还是可以的8)) { qh(wz*2);//如果有被包含,也要去左边的小块看看 } }
嘿嘿简单吧^-^,啊我忘记讲建树了QAQ。没事马上就讲。
建树嘛,直接递归就好了,每次把这个区间分成两份,一直分到l和r一样(就是说这是一个点的值,要输入了),然后获取两个点的值之后就可以向上递归~嗯嗯对,一直递归就建好了,相信大家都有这个写递归的能力,但要实在不会就看看下面的代码吧(温馨提示:线段树用来建树的数组要开到节点数的4倍大,至于为什么我也忘了……):
struct hehe { long long l,r,shu,f; }tree[400005];//tree结构体~ long long mid; void js(long long ll,long long rr,long long wz)//js,建树的意思。ll和rr分别是左右边界,wz就是他的编号 { tree[wz].l=ll;//左边界是ll tree[wz].r=rr;//右边界是rr if(ll==rr)//左右边界一致,这是一个点 { scanf("%lld",&tree[wz].shu); return; } long long mid=(ll+rr)/2;//如果这不是一个点,肯定能分成两部分 js(ll,mid,wz*2);//左边 js(mid+1,rr,wz*2+1);//右边 tree[wz].shu=tree[wz*2].shu+tree[wz*2+1].shu;//加起来 }
看,是不是非常简单,接下来为了学的更深一点点,要把题目加难了(是的还要加难,但我相信你们一定可以学会的)
有一个序列,要做到区间增加区间求和,该怎么做呢?
啊这,这个是线段树里最难的一部分(起码我觉得最难),直接下放显然不太现实……,会浪费掉很多不必要的时间。正所谓科技发展在于懒人~,其实我们也可以在不影响结果的情况下偷个懒是吧QWQ。
就像加一样~我知道这两个序列的和,我就没必要去求所有单个序列是多少,加法和这个差不多,我们可以看看有那个序列是全都要加的,然后直接算出它加完之后的值。这样求这个值的时候肯定是不影响计算的(偷懒成功!)。但这时就会有一些同学吐槽:”啊你这个不严谨啊,如果下一次求和是求这个序列的一部分怎么办呢?”其实呢,我刚才也写了是吧……
其实我们也可以在不影响结果的情况下偷个懒是吧
我刚才说的是不影响,但这个操作明显影响了,在哪里影响了呢?就是只加了一个总序列,没有让他的一部分加上(偷懒失败QWQ)。但好像根本没有必要调整他的一部分啊,因为目前根本用不到,没说让做的事情我们还要去做不是浪费时间吗?但又不能不加,怎么办呢?我们可以设定一个懒标记,表示它之前被加了多少,这样呢,每次要求和的时候只要下放这个懒标记,并且加上该加的数,就能达到不说不做,最大程度优化时间,还保证正确hhhhh(偷懒成功)
至于代码的实现也是非常的简单:
void xf(long long wz) { tree[wz*2].shu+=(tree[wz*2].r-tree[wz*2].l+1)*tree[wz].f;//每个长度单位都加上tree[wz].f,总值就增加了(tree[wz*2].r-tree[wz*2].l+1)*tree[wz].f tree[wz*2].f+=tree[wz].f;//之前每一个长度单位要加tree[wz*2].f,现在发现它还要加上tree[wz].f,就把他们两个加起来就好了嘛,很容易的。 tree[wz*2+1].shu+=(tree[wz*2+1].r-tree[wz*2+1].l+1)*tree[wz].f; tree[wz*2+1].f+=tree[wz].f;//同理 tree[wz].f=0;//这里已经加过了,要清零的,就像转账一样,你给另一个人转了钱,你钱就没了。 }
上方是下放懒标记的代码,下方是区间修改代码:
void xg(long long wz) { if(tree[wz].l>=a&&tree[wz].r<=b) { tree[wz].shu+=(tree[wz].r-tree[wz].l+1)*c;//更改值 tree[wz].f+=c;//增加懒标记 return; }//啊后面的都说过,不打了… tree[wz].shu+=(min(tree[wz].r,b)-max(tree[wz].l,a)+1)*c; long long mid=(tree[wz].l+tree[wz].r)/2; if(b>mid) { xg(wz*2+1); } if(a<=mid) { xg(wz*2); } }
至于加了懒标记以后别的代码也是要动的。
比如在原来的求和代码里多了一个xf(wz);
至于加在哪里同学们自己斯烤吧(我太仁慈了)
啊我再放一道线段树模板题,大家可以秒掉来吊打我:模板题传送们
好了我觉得我讲完了,如果有什么不好或漏掉的地方大家可以及时评论,我会更改的。