这道题来回改了好多遍。。线段树的题还是要多练。
主要是lazy的用法,牵扯到两种lazy标记,其中替换标记如果在增减标记的后边,那么要清空增减标记。
权当重新理了一遍线段树的模板了。。不过这个不用query
描述
小Hi和小Ho都是游戏迷,“模拟都市”是他们非常喜欢的一个游戏,在这个游戏里面他们可以化身上帝模式,买卖房产。
在这个游戏里,会不断的发生如下两种事件:一种是房屋自发的涨价或者降价,而另一种是政府有关部门针对房价的硬性调控。房价的变化自然影响到小Hi和小Ho的决策,所以他们希望能够知道任意时刻某个街道中所有房屋的房价总和是多少——但是很不幸的,游戏本身并不提供这样的计算。不过这难不倒小Hi和小Ho,他们将这个问题抽象了一下,成为了这样的问题:
小Hi和小Ho所关注的街道的长度为N米,从一端开始每隔1米就有一栋房屋,依次编号为0..N,在游戏的最开始,每栋房屋都有一个初始价格,其中编号为i的房屋的初始价格为p_i,之后共计发生了M次事件,所有的事件都是对于编号连续的一些房屋发生的,其中第i次事件如果是房屋自发的涨价或者降价,则被描述为三元组(L_i, R_i, D_i),表示编号在[L_i, R_i]范围内的房屋的价格的增量(即正数为涨价,负数为降价)为D_i;如果是政府有关部门针对房价的硬性调控,则被描述为三元组(L_i, R_i, V_i),表示编号在[L_i, R_i]范围内的房屋的价格全部变为V_i。而小Hi和小Ho希望知道的是——每次事件发生之后,这个街道中所有房屋的房价总和是多少。
输入
每个测试点(输入文件)有且仅有一组测试数据。
每组测试数据的第1行为两个整数N、M,分别表示街道的长度和总共发生的事件数。
每组测试数据的第2行为N+1个整数,其中第i个整数位p_i,表示编号为i的房屋的初始价格。
每组测试数据的第3-M+2行,按照发生的时间顺序,每行描述一个事件,如果该行描述的事件为,“房屋自发的涨价或者降价”,则该行为4个整数0, L_i, R_i, D_i,意义如前文所述;如果该行描述的事件为“政府有关部门针对房价的硬性调控”,则该行为4个整数1, L_i, R_i, V_i,意义如前文所述。
对于100%的数据,满足N<=10^5,1<=p_i, |D_i|, V_i<=10^4,0<=l_i<r_i<=n。<>
对于100%的数据,满足在任意时刻,任何房屋的价格都处于[1, 10^4]内。
输出
对于每组测试数据,输出M行,其中第i行为一个整数Ans_i,表示第i次事件发生之后,这个街道中所有房屋的房价总和。
10 6 3195 2202 4613 3744 2892 4858 619 5079 9478 7366 8942 0 1 6 886 1 0 2 9710 1 0 10 7980 0 4 9 -7594 0 2 8 1581 0 4 4 -1010样例输出
58304 75652 87780 42216 53283 52273
具体看代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
int A[maxn];
struct node
{
int sum;
int seg[3]={0};
}tree[maxn<<2];
void pushup(int rt)
{
tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void push_down(int root,int len)
{//顺序必须是先替换,再增添。因为先增添后替换的那种情况之前已经舍弃了
if(tree[root].seg[2])//替换
{
tree[root<<1].seg[2]=tree[root<<1|1].seg[2]=tree[root].seg[2];
tree[root<<1].seg[1]=tree[root<<1|1].seg[1]=0;//清0
tree[root<<1].sum=(len-(len>>1))*tree[root].seg[2];
tree[root<<1|1].sum=(len>>1)*tree[root].seg[2];
tree[root].seg[2]=0;//lazy标记归0
}
if(tree[root].seg[1])
{
tree[root<<1].seg[1]+=tree[root].seg[1];//叠加
tree[root<<1|1].seg[1]+=tree[root].seg[1];
tree[root<<1].sum+=(len-(len>>1))*tree[root].seg[1];
tree[root<<1|1].sum+=(len>>1)*tree[root].seg[1];
tree[root].seg[1]=0;
}
}
void build(int root,int l,int r)
{
if(l==r)
{
tree[root].sum=A[r];
}
else
{
int mid=(l+r)>>1;
build(root*2,l,mid);
build(root*2+1,mid+1,r);
pushup(root);
}
}
void update(int logal,int root,int L,int R,int l,int r,int val)
{
if(L<=l&&R>=r)
{
if(logal==1)//替换
{
tree[root].sum=(r-l+1)*val;
tree[root].seg[2]=val;
tree[root].seg[1]=0;//清空增添标记,这时说明是先增添后替换,所以按照替换的来
}
else
{
tree[root].sum+=(r-l+1)*val;
tree[root].seg[1]+=val;
}
return;
}
push_down(root,r-l+1);
int mid=(l+r)>>1;
if(L<=mid)
update(logal,root<<1,L,R,l,mid,val);
if(R>mid)
update(logal,root<<1|1,L,R,mid+1,r,val);
pushup(root);
}
int main()
{
int n,m;
int a,b,c,d;
scanf("%d%d",&n,&m);
for(int i=1;i<=n+1;i++)
{
cin>>A[i];
}
build(1,1,n+1);
for(int i=1;i<=m;i++)
{
cin>>a>>b>>c>>d;
update(a,1,b+1,c+1,1,n+1,d);
cout<<tree[1].sum<<endl;
}
return 0;
}