线段树技巧

线段树技巧

Part 1: 简单的常数优化

1.非递归线段树

你可以不用zkw,只用每次单点操作时直接像打二分样写下去就好了

以单点求和为例

int Que(int p,int l,int r,int x){
    if(l==r) return Sum[p];
    int mid=(l+r)>>1;
    Down(p);
    if(x<=mid) return Que(p<<1,l,mid,x);
    else return Que(p<<1|1,mid+1,r,x);
}//递归
int Que(int x){
    int p=1,l=1,r=n,mid;
    while(l!=r) {
        mid=(l+r)>>1;
        if(x<=mid) r=mid,p=p<<1|1;
        else l=mid+1,p=p<<1;
    }
    return Sum[p];
}//非递归

当然你可以花一些时间之间去学zkw

2.标记永久化

当操作只有区间修改和单点查询时,可以用(事实上这是一种很有通用的技巧)

查询时直接将遇到的点的Sum值累和过来

void Upd(int p,int l,int r,int ql,int qr,int x){
    if(l==ql&&r==qr) {
		Sum[p]+=(r-l+1)*x;
		tag[p]+=x;
		return;
        
	}
    Down(p);
	int mid=(l+r)>>1;
	if(qr<=mid) Upd(p<<1,l,mid,ql,qr,x);
	else if(ql>mid) Upd(p<<1|1,mid+1,r,ql,qr,x);
	else Upd(p<<1,l,mid,ql,mid,x),Upd(p<<1|1,mid+1,r,mid+1,qr,x);
	Sum[p]=Sum[p<<1]+Sum[p<<1|1];
}
int Que(int p,int l,int r,int x){
    if(l==r) return Sum[p];
    int mid=(l+r)>>1;
    Down(p);
    if(x<=mid) return Que(p<<1,l,mid,x);
    else return Que(p<<1|1,mid+1,r,x);
}
//普通标记法


void Upd(int p,int l,int r,int ql,int qr,int x){
    if(l==ql&&r==qr) {
		Sum[p]+=(r-l+1)*x;
		return;
	}
	Down(p);
	int mid=(l+r)>>1;
	if(qr<=mid) Upd(p<<1,l,mid,ql,qr,x);
	else if(ql>mid) Upd(p<<1|1,mid+1,r,ql,qr,x);
	else Upd(p<<1,l,mid,ql,mid,x),Upd(p<<1|1,mid+1,r,mid+1,qr,x);
}

void Que(int p,int l,int r,int x,int &res){
	res+=Sum[p];
    if(l==r) return;
    int mid=(l+r)>>1;
    if(x<=mid) Que(p<<1,l,mid,x,res);
    else Que(p<<1|1,mid+1,r,x,res);
}
//标记永久化
//注意不要Up

\[\ \]

\[\ \]

\[\ \]

Part 2:神奇的用法

(当你做了一道序列上的dp题,发现自己只会打暴力怎么办?线段树不就是了!)

1.区间取模操作

将一个区间\([L,R]\)的数对\(x\)取模

发现性质当\(a[i]>=x\)时, \(a[i] \mod x \leq \frac {a[i]}2\)

因为当\(a[i] < x*2\)时,\(a[i] \mod x= a[i]-x < a[i]/2\)

\(a[i] \geq x*2\)时,\(a[i] \mod x<=x<=\frac{a[i]}2\)

所以直接存一下最大值,大于等于x时就递归下去,复杂度\(log^2 n\)

void Upd(int p,int l,int r,int ql,int qr,int x) {
	if(l==r) {
		ma[p]%=x;
		s[p]=is[ma[p]];
		return;
	}
	int mid=(l+r)>>1;
	if(ql==l&&qr==r) {
		if(ma[p<<1]>=x) Upd(p<<1,l,mid,ql,mid,x);
		if(ma[p<<1|1]>=x) Upd(p<<1|1,mid+1,r,mid+1,qr,x);
	} else {
		if(qr<=mid) Upd(p<<1,l,mid,ql,qr,x);
		else if(ql>mid) Upd(p<<1|1,mid+1,r,ql,qr,x);
		else Upd(p<<1,l,mid,ql,mid,x),Upd(p<<1|1,mid+1,r,mid+1,qr,x);
	}
	ma[p]=max(ma[p<<1],ma[p<<1|1]);
	s[p]=s[p<<1]+s[p<<1|1];
}

2.内外标记结合

当你的标记在线段树上不好处理时,可以选择在整个序列上操作,把标记存在外面

\(Set(a[1..n])=x-a[1..n]\)

可以在整个序列外面记录,将整个序列符号取反,再加x

3.整体偏移大法

当你遇到这样一个dp

\[dp[i][j]=dp[i-1][j-x] \]

如何搞?直接在序列外面记录偏移量\(d\),保留线段树中原来节点的权值

其实这时一个映射的思想

初始情况下$i \rightarrow i $

偏移后\(i \rightarrow i+x\)

但是偏移后值域改变?

不,你只用在刚开始时整体值域弄大一点就好了

真的不行我们还有动点线段树啊

事实上这只是一种简单的偏移,它还可以完成更多操作

\(dp[i][j]=dp[i-1][x-j]\)

只用在偏移上乘上一个系数就好了

注意偏移后,查询\(L,R\)时,要带入系数算出原来的位置,解个方程就好了

是不是很\(easy\)!

HDU6728

这题我就是暴力搞过去的


const int N=3e3+10,K=230,P=1e9+7;

const ll L=-1e11,R=1e11;


int n;
ll c[N];

int cnt;
int s[N*K],t1[N*K],t2[N*K];
int ls[N*K],rs[N*K];

void Down(int p,ll l,ll r){
    ll mid=(l+r)>>1;
    if(~t1[p]) {
        t1[rs[p]]=t1[p];
        t1[ls[p]]=t1[p];
        t2[ls[p]]=t2[rs[p]]=0;
        s[ls[p]]=1ll*t1[p]*((mid-l+1)%P)%P;
        s[rs[p]]=1ll*t1[p]*((r-mid)%P)%P;
        t1[p]=-1;
    }
    if(t2[p]) {
        (t2[rs[p]]+=t2[p])%=P;
        (t2[ls[p]]+=t2[p])%=P;
        (s[ls[p]]+=1ll*t2[p]*((mid-l+1)%P)%P)%=P;
        (s[rs[p]]+=1ll*t2[p]*((r-mid)%P)%P)%=P;
        t2[p]=0;
    }
}

void Set(int p,ll l,ll r,ll ql,ll qr,int x){
    x%=P;
    if(ql>qr) return;
    if(l==ql&&r==qr){
        s[p]=1ll*(r-l+1)%P*x%P;
        t1[p]=x;
        t2[p]=0;
        return;
    }
    if(!ls[p]) ls[p]=++cnt;
    if(!rs[p]) rs[p]=++cnt;
    Down(p,l,r);
    ll mid=(l+r)>>1;
    if(qr<=mid) Set(ls[p],l,mid,ql,qr,x);
    else if(ql>mid) Set(rs[p],mid+1,r,ql,qr,x);
    else Set(ls[p],l,mid,ql,mid,x),Set(rs[p],mid+1,r,mid+1,qr,x);
    s[p]=(s[ls[p]]+s[rs[p]])%P;
}

void Upd(int p,ll l,ll r,ll ql,ll qr,int x){
    if(ql>qr) return;
    if(l==ql&&r==qr){
         (s[p]+=(r-l+1)%P*x%P)%=P;
         (t2[p]+=x)%=P;
         return;
    }
    if(!ls[p]) ls[p]=++cnt;
    if(!rs[p]) rs[p]=++cnt;
    Down(p,l,r);
    ll mid=(l+r)>>1;
    if(qr<=mid) Upd(ls[p],l,mid,ql,qr,x);
    else if(ql>mid) Upd(rs[p],mid+1,r,ql,qr,x);
    else Upd(ls[p],l,mid,ql,mid,x),Upd(rs[p],mid+1,r,mid+1,qr,x);
    s[p]=(s[ls[p]]+s[rs[p]])%P;
}

ll Que(int p,ll l,ll r,ll ql,ll qr) {
    if(ql>qr) return 0;
    if(l==ql&&r==qr) return s[p]%=P;
    if(!ls[p]) ls[p]=++cnt;
    if(!rs[p]) rs[p]=++cnt;
    Down(p,l,r);
    ll mid=(l+r)>>1;
    if(qr<=mid) return Que(ls[p],l,mid,ql,qr);
    else if(ql>mid) return Que(rs[p],mid+1,r,ql,qr);
    else return (Que(ls[p],l,mid,ql,mid)+Que(rs[p],mid+1,r,mid+1,qr))%P;
}


int main(){
    rep(kase,1,rd()) {
        cnt=1;
        memset(ls,0,sizeof ls);
        memset(rs,0,sizeof rs);
        memset(t2,0,sizeof t2);
        memset(t1,-1,sizeof t1);
        memset(s,0,sizeof s);
        n=rd();
        rep(i,2,n) c[i]=rd();
        Upd(1,L,R,1,R,1);
        ll fl=1,d=0;
        rep(i,2,n) {
            ll t1,t2;
            t1=(1-d)*fl,t2=(c[i]-d)*fl;
            ll sum;
            if(fl==1) sum=Que(1,L,R,t1,t2);
            else sum=Que(1,L,R,t2,t1);
            ll t=Que(1,L,R,t2,t2);
        
            ll tmp=-d/fl;
            Set(1,L,R,tmp,tmp,sum);
            t2=(c[i]-1-d)*fl;
            if(fl==1) {
                Set(1,L,R,L,tmp-1,0);
                Set(1,L,R,t2+1,R,0);
            } else {
                Set(1,L,R,L,t2-1,0);
                Set(1,L,R,tmp+1,R,0);
            }
            
            if(fl==1) {
                Upd(1,L,R,t1,t2,t);
            } else Upd(1,L,R,t2,t1,t);
            fl=-fl;
            d=c[i]-d;
            if(d>=R||d<=L) while(1);
        }
        ll ans=Que(1,L,R,L,R)%P;
        ans=(ans%P+P)%P;
        printf("%lld\n",ans);
    }
}

\[\ \]

4.区间取\(min\),单点查询

用一个标记记录即可,可以和其他多种标记(如:加法,赋值)

问题都出在标记的合并问题上,这个只能根据具体情况来模拟了

5.区间对于\(a\)\(min\),区间查询

每个区间记录两个值\((x,y,c)\)表示最大值,严格次大值,最大值出现的次数

考虑多种更新情况

i. \(x\leq a\)跳过

ii.\(y<a<x\)只有最大值会被影响到,直接按照个数更新即可

tip:必须是\(y<a\)才能更新,因更新之后新的\(x=a\),如果\(y=a\),那么最大值的个数无法直接求出

iii.\(a\leq y\) 此时需要递归到子树取更新

可以证明这样递归的次数是远达不到\(O(m\log^2n)\)

如果不包含其他修改,可以被证明是\(O(m\log n)\)

这样的做法对于取\(min\),甚至是同时取\(min\),取\(max\)都可以写

下面是标准的板子题

HDU-5306 Gorgeous Sequence

const int N=1e6+10;
const int INF=(1ll<<31)-1;


int n,m;
int a[N];
struct Node{
	ll sum;
	int x,y,c;
	Node operator + (const Node __) const {
		Node res;
		res.sum=sum+__.sum;
		res.x=max(x,__.x);
		res.c=0;
		if(x==res.x) res.c+=c;
		if(__.x==res.x) res.c+=__.c;
		res.y=-1;
		if(x<res.x) cmax(res.y,x);
		if(y<res.x) cmax(res.y,y);
		if(__.x<res.x) cmax(res.y,__.x);
		if(__.y<res.x) cmax(res.y,__.y);
		return res;
	}
}s[N<<2];

int t[N<<2];
void Build(int p,int l,int r) {
	t[p]=INF;
	if(l==r) {
		s[p]=(Node){a[l],a[l],-1,1};
		return;
	}
	int mid=(l+r)>>1;
	Build(p<<1,l,mid),Build(p<<1|1,mid+1,r);
	s[p]=s[p<<1]+s[p<<1|1];
}

void Down(int p) {
	if(t[p]==INF) return;
	rep(i,p<<1,p<<1|1) if(s[i].x>t[p]) {
		s[i].sum-=1ll*(s[i].x-t[p])*s[i].c,s[i].x=t[p];
		cmin(t[i],t[p]); // 注意只有x>a 时,才能够下传标记,否则会挂(当然如果特判了就另当别论)
	}
	t[p]=INF;
}

void Upd(int p,int l,int r,int ql,int qr,int x) {
	if(ql<=l && r<=qr) {
		if(s[p].x<=x) return; 
		if(s[p].y<x && x<s[p].x) { // 这种情况需要更新
			cmin(t[p],x);
			s[p].sum-=1ll*(s[p].x-x)*s[p].c;
			s[p].x=x;
			return;
		}
	}
	int mid=(l+r)>>1;
	Down(p);
	if(ql<=mid) Upd(p<<1,l,mid,ql,qr,x);
	if(qr>mid) Upd(p<<1|1,mid+1,r,ql,qr,x);
	s[p]=s[p<<1]+s[p<<1|1];
}

Node Que(int p,int l,int r,int ql,int qr) {
	if(ql==l && r==qr) return s[p];
	Down(p);
	int mid=(l+r)>>1;
	if(qr<=mid) return Que(p<<1,l,mid,ql,qr);
	else if(ql>mid) return Que(p<<1|1,mid+1,r,ql,qr);
	else return Que(p<<1,l,mid,ql,mid)+Que(p<<1|1,mid+1,r,mid+1,qr);
}

int main(){
	rep(kase,1,rd()) {
		n=rd(),m=rd();
		rep(i,1,n) a[i]=rd();
		Build(1,1,n);
		rep(i,1,m) {
			int opt=rd(),l=rd(),r=rd();
			if(opt==0) Upd(1,1,n,l,r,rd());
			else {
				Node res=Que(1,1,n,l,r);
				if(opt==1) printf("%d\n",res.x);
				else printf("%lld\n",res.sum);
			}
		}
	}
}

\[\ \]

6.\(gcd\)和区间更新

例子:

1.求\(gcd(a[l],a[l+1],..a[r])\)

2.对于区间\(l,r\),增加\(x\)

\[\ \]

利用\(gcd(a,b)=gcd(a,b-a)\),序列差分之后,这个问题可以转化为单点修改问题

这个思想适用于这一类区间求\(gcd\)的问题

\[\ \]

7.历史版本最大值

即考虑之前 (之前每次操作过后这段区间的\(max\)) 的\(max\)

考虑对于每一种标记记录一个\(max\),表示 这个标记在历史版本的最大值,更新时就可以同步更新了

这里有一个复杂的多标记结合+历史版本最大值的题

BZOJ3064CPU监控

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); }

char IO;
int rd(){
	int s=0,f=0;
	while(!isdigit(IO=getchar())) if(IO=='-') f=1;
	do s=(s<<1)+(s<<3)+(IO^'0');
	while(isdigit(IO=getchar()));
	return f?-s:s;
}

const int N=1e5+10;
const ll INF=1e18;

int n,m;
char opt[2];
ll a[N],s1[N<<2],s2[N<<2],t1[N<<2],t2[N<<2],g1[N<<2],g2[N<<2];
void Up(int p) { s1[p]=max(s1[p<<1],s1[p<<1|1]),s2[p]=max(s2[p<<1],s2[p<<1|1]); }

void Build(int p,int l,int r) {
	if(l==r) return void(s1[p]=s2[p]=a[l]);
	t1[p]=t2[p]=g1[p]=g2[p]=-INF;
	int mid=(l+r)>>1;
	Build(p<<1,l,mid),Build(p<<1|1,mid+1,r),Up(p);
}

void Mark(int p,ll x,int k) {
	if(k==1) {
		s1[p]+=x,cmax(s2[p],s1[p]);
		if(t2[p]!=-INF) t2[p]+=x,cmax(g2[p],t2[p]);
		else t1[p]=(t1[p]==-INF)?x:t1[p]+x,cmax(g1[p],t1[p]);
	} else {
		t1[p]=-INF,t2[p]=x,cmax(g2[p],t2[p]);
		s1[p]=x,cmax(s2[p],s1[p]);
	}
}

void Down(int p) {
	if(t1[p]==-INF && t2[p]==-INF) return;
	rep(i,p<<1,p<<1|1) {
		if(t1[i]!=-INF) cmax(g1[i],g1[p]+t1[i]);
		else if(t2[i]!=-INF) cmax(g2[i],g1[p]+t2[i]);
		else cmax(g1[i],g1[p]);
		cmax(s2[i],s1[i]+g1[p]),cmax(g2[i],g2[p]),cmax(s2[i],g2[p]);
	}
	if(t2[p]!=-INF) Mark(p<<1,t2[p],2),Mark(p<<1|1,t2[p],2);
	else Mark(p<<1,t1[p],1),Mark(p<<1|1,t1[p],1);
	t1[p]=t2[p]=g1[p]=g2[p]=-INF;
}

void Upd(int p,int l,int r,int ql,int qr,int x,int k) {
	if(ql<=l && r<=qr) return Mark(p,x,k);
	Down(p);
	int mid=(l+r)>>1;
	if(ql<=mid) Upd(p<<1,l,mid,ql,qr,x,k);
	if(qr>mid) Upd(p<<1|1,mid+1,r,ql,qr,x,k);
	Up(p);
}

ll Que(int p,int l,int r,int ql,int qr,int k) {
	if(ql<=l && r<=qr) return k==1?s1[p]:s2[p];
	Down(p);
	int mid=(l+r)>>1;
	ll res=-INF;
	if(ql<=mid) cmax(res,Que(p<<1,l,mid,ql,qr,k));
	if(qr>mid) cmax(res,Que(p<<1|1,mid+1,r,ql,qr,k));
	return res;
}

int main(){
	n=rd();
	rep(i,1,n) a[i]=rd();
	m=rd();
	Build(1,1,n);
	rep(i,1,m) {
		scanf("%s",opt);
		int l=rd(),r=rd();
		if(opt[0]=='Q') printf("%lld\n",Que(1,1,n,l,r,1));
		else if(opt[0]=='A') printf("%lld\n",Que(1,1,n,l,r,2));
		else Upd(1,1,n,l,r,rd(),opt[0]=='P'?1:2);
	}
}

\[\ \]

原文地址:https://www.cnblogs.com/chasedeath/p/11446631.html