线段树模板练习题

1.[NOIP2012]借教室

传送门

题目描述

    在大学期间，经常需要租借教室。大到院系举办活动，小到学习小组自习讨论，都需要向学校申请借教室。教室的大小功能不同，借教室人的身份不同，借教室的手续也不一样。

    面对海量租借教室的信息，我们自然希望编程解决这个问题。

    我们需要处理接下来n天的借教室信息，其中第i天学校有ri个教室可供租借。共有m份订单，每份订单用三个正整数描述，分别为dj, sj, tj，表示某租借者需要从第sj天到第tj天租借教室（包括第sj天和第tj天），每天需要租借dj个教室。

    我们假定，租借者对教室的大小、地点没有要求。即对于每份订单，我们只需要每天提供dj个教室，而它们具体是哪些教室，每天是否是相同的教室则不用考虑。

    借教室的原则是先到先得，也就是说我们要按照订单的先后顺序依次为每份订单分配教室。如果在分配的过程中遇到一份订单无法完全满足，则需要停止教室的分配，通知当前申请人修改订单。这里的无法满足指从第sj天到第tj天中有至少一天剩余的教室数量不足dj个。

    现在我们需要知道，是否会有订单无法完全满足。如果有，需要通知哪一个申请人修改订单。

输入描述:

第一行包含两个正整数n, m，表示天数和订单的数量。

第二行包含n个正整数，其中第i个数为ri，表示第i天可用于租借的教室数量。

接下来有m行，每行包含三个正整数dj, sj, tj，表示租借的数量，租借开始、结束分别在第几天。

每行相邻的两个数之间均用一个空格隔开。天数与订单均用从1开始的整数编号。

输出描述:

如果所有订单均可满足，则输出只有一行，包含一个整数0。否则（订单无法完全满足）输出两行，第一行输出一个负整数-1，第二行输出需要修改订单的申请人编号。

示例1

输入：

4 3
2 5 4 3
2 1 3
3 2 4
4 2 4

输出：

-1
2

 _____________________________________________________________________________

模板题。简化一下，就是给定一个长度为n的序列，每一次操作将[s,t]的数-d（代表借出去了），查询[s,t]的数的最小值是否小于0。小于0则代表不能满足订单。

 所以可用线段树维护区间最小值，操作为区间修改，需要懒标记。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int a[N],n,m;
struct tree{
    int l,r,v,c;
}T[N*4];
void up(int p)
{
    T[p].v=min(T[p<<1].v,T[p<<1|1].v);
}
void down(int p)
{
    if(T[p].c){
        T[p<<1].c+=T[p].c;
        T[p<<1|1].c+=T[p].c;
        T[p<<1].v-=T[p].c;
        T[p<<1|1].v-=T[p].c;
        T[p].c=0;
    }
}
void build_tree(int p,int l,int r)
{
    T[p].l=l,T[p].r=r,T[p].c=0;
    if(l==r){
        T[p].v=a[l];
        return;
    }
    int m=l+r>>1;
    build_tree(p<<1,l,m),build_tree(p<<1|1,m+1,r);
    up(p);
}
void modify(int p,int l,int r,int d)
{
    if(T[p].l>=l&&T[p].r<=r){
        T[p].c+=d;
        T[p].v-=d;
        return;
    }
    down(p);
    if(T[p<<1].r>=l) modify(p<<1,l,r,d);
    if(T[p<<1|1].l<=r) modify(p<<1|1,l,r,d);
    up(p);
}
int query(int p,int l,int r)
{
    if(T[p].l>=l&&T[p].r<=r) return T[p].v;
    down(p);
    int res=INT_MAX;
    if(T[p<<1].r>=l) res=min(res,query(p<<1,l,r));
    if(T[p<<1|1].l<=r) res=min(res,query(p<<1|1,l,r));
    return res;
}
int main()
{
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
    for(int i=1;i<=n;++i) cin>>a[i];
    build_tree(1,1,n);
    for(int i=1;i<=m;++i){
        int d,s,t;
        cin>>d>>s>>t;
        modify(1,s,t,d);
        if(query(1,s,t)<0){
            cout<<-1<<endl<<i;
            return 0;
        }
    }
    cout<<0;
	return 0;
}

2.数据结构

传送门

题目描述

qn姐姐最好了~

qn姐姐给你了一个长度为n的序列还有m次操作让你玩，

1 l r 询问区间[l,r]内的元素和

2 l r 询问区间[l,r]内的元素的平方 和

3 l r x 将区间[l,r]内的每一个元素都乘上x

4 l r x 将区间[l,r]内的每一个元素都加上x

输入描述:

第一行两个数n,m

接下来一行n个数表示初始序列

就下来m行每行第一个数为操作方法opt，

若opt=1或者opt=2，则之后跟着两个数为l,r

若opt=3或者opt=4，则之后跟着三个数为l,r,x

操作意思为题目描述里说的

输出描述:

对于每一个操作1,2，输出一行表示答案

示例1

输入：

5 6
1 2 3 4 5
1 1 5
2 1 5
3 1 2 1
4 1 3 2
1 1 4
2 2 3

输出：

15
55
16
41

备注:

对于100%的数据 n=10000,m=200000 (注意是等于号)

保证所有询问的答案在long long 范围内

 _____________________________________________________________________________

线段树需要维护和，平方和，加数add，乘数mul；

是选择先加再乘，还是先乘再加，关键在于在做down操作的时候如何更新子节点方便。 

1.先add再mul，即标记代表把区间每一个数先加add，再乘mul

更新后很难完成形式统一，不好表示。

2.先mul再add，即标记代表把区间每一个数先乘mul，再加add

 

故选则第二种。

更新函数：

void eval(tree &t,ll add,ll mul)
{
	t.s2=mul*mul*t.s2+add*add*(t.r-t.l+1)+2*mul*add*t.s;
	t.s=t.s*mul+add*(t.r-t.l+1);
	t.mul=mul*t.mul;
	t.add=t.add*mul+add;
}

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=10010; 
int n,a[N];
struct tree{
	int l,r;
	ll s,s2,add,mul;
}T[4*N];
void up(int p)
{
	T[p].s=T[p<<1].s+T[p<<1|1].s;
	T[p].s2=T[p<<1].s2+T[p<<1|1].s2;
}
void eval(tree &t,ll add,ll mul)
{
	t.s2=mul*mul*t.s2+add*add*(t.r-t.l+1)+2*mul*add*t.s;
	t.s=t.s*mul+add*(t.r-t.l+1);
	t.mul=mul*t.mul;
	t.add=t.add*mul+add;
}
void down(int p)
{
	eval(T[p<<1],T[p].add,T[p].mul);
	eval(T[p<<1|1],T[p].add,T[p].mul);
	T[p].add=0,T[p].mul=1;
}
void modify(int p,int l,int r,int add,int mul)
{
	if(T[p].l>=l&&T[p].r<=r){
		eval(T[p],add,mul);
		return;
	}
	down(p);
	if(T[p<<1].r>=l) modify(p<<1,l,r,add,mul);
	if(T[p<<1|1].l<=r) modify(p<<1|1,l,r,add,mul);
	up(p);
}
ll query(int p,int l,int r)
{
	if(T[p].l>=l&&T[p].r<=r) return T[p].s;
	down(p);
	ll res=0;
	if(T[p<<1].r>=l) res+=query(p<<1,l,r); 
	if(T[p<<1|1].l<=r) res+=query(p<<1|1,l,r);
	return res;
}
ll query2(int p,int l,int r)
{
	if(T[p].l>=l&&T[p].r<=r) return T[p].s2;
	down(p);
	ll res=0;
	if(T[p<<1].r>=l) res+=query2(p<<1,l,r); 
	if(T[p<<1|1].l<=r) res+=query2(p<<1|1,l,r);
	return res;
}
void build_tree(int p,int l,int r)
{
	T[p]={l,r,0,0,0,1};
	if(l==r){
		T[p].s=a[l],T[p].s2=1ll*a[l]*a[l];
		return;
	}
	int m=l+r>>1;
	build_tree(p<<1,l,m),build_tree(p<<1|1,m+1,r);
	up(p);
}
int main()
{
	int _;
	cin>>n>>_;
	for(int i=1;i<=n;++i) cin>>a[i];
	build_tree(1,1,n);
	while(_--){
		int op,l,r,x;
		cin>>op>>l>>r;
		if(op==1) cout<<query(1,l,r)<<endl;
		else if(op==2) cout<<query2(1,l,r)<<endl;
		else if(op==3){
			cin>>x;
			modify(1,l,r,0,x);
		}
		else{
			cin>>x;
			modify(1,l,r,x,1);
		}
	}
	return 0;
}