您的位置:首页 > 其它

hdu 1698 线段树的区间修改 模板~~

2015-07-25 17:48 447 查看
把给定区间内的数全部更改为另一个数,线段树区间更新 模版,参考别人的代码

#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#define inf 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;

inline int in()
{
int res=0;char c;
while((c=getchar())<'0' || c>'9');
while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
return res;
}
struct st
{
int l,r,sum;
int lazy,tag;
}a[100010<<2]; //记得开4倍,老忘记

void build(int l,int r,int i)//递归的建树就行
{
a[i].l=l;
a[i].r=r;
a[i].tag=0;
a[i].lazy=0;
if(l==r)
{
a[i].sum=1;
return;
}
int mid=(l+r)>>1;
build(l,mid,i<<1);
build(mid+1,r,i<<1|1);
a[i].sum=a[i<<1].sum+a[i<<1|1].sum;
}

void update(int l,int r,int x,int i)
{
if(a[i].l==l && a[i].r==r) //成端更新
{
a[i].lazy=1;
a[i].tag=x;
a[i].sum=(r-l+1)*x;
return;
}
int mid=(a[i].l+a[i].r)>>1;
if(a[i].lazy==1)    //当前节点有lazy标记
{
a[i].lazy=0;
update(a[i].l,mid,a[i].tag,i<<1);  //向下更新左半边,传参数应该传tag
update(mid+1,a[i].r,a[i].tag,i<<1|1);//向下更新右半边
a[i].tag=0;                         //设为零
}
if(r<=mid)update(l,r,x,i<<1);      //要更新的区间在当前区间的左边
else if(l>mid)update(l,r,x,i<<1|1);//在右边
else                                  //两边都在
{
update(l,mid,x,i<<1);           //传mid
update(mid+1,r,x,i<<1|1);
}
a[i].sum=a[i<<1].sum+a[i<<1|1].sum; //更新
}

int main()
{
int T=in();
int iCase=1;
while(T--)
{
int n=in();
build(1,n,1);
int m=in();
int x,y,z;
while(m--)
{
x=in(),y=in(),z=in();
update(x,y,z,1);
}
printf("Case %d: The total value of the hook is %d.\n",iCase++,a[1].sum);
}
return 0;
}


下面贴自己写的

#include<bitset>
#include<map>
#include<vector>
#include<cstdio>
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<stack>
#include<queue>
#include<set>
#define inf 0x3f3f3f3f
#define mem(a,x) memset(a,x,sizeof(a))

using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
inline int in()
{
int res=0;
char c;
while((c=getchar())<'0' || c>'9');
while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
return res;
}
struct st
{
int l,r;
int lazy,sum;
}a[400020];

void PushUp(int i)
{
a[i].sum=a[i<<1].sum+a[i<<1|1].sum;
}

void PushDown(int i)
{
if(a[i].lazy)
{
a[i<<1].lazy=a[i].lazy;
a[i<<1|1].lazy=a[i].lazy;
a[i<<1].sum=(a[i<<1].r-a[i<<1].l+1)*a[i].lazy;
a[i<<1|1].sum=(a[i<<1|1].r-a[i<<1|1].l+1)*a[i].lazy;
a[i].lazy=0;
}
}
void build(int l,int r,int i)
{
a[i].l=l,a[i].r=r;
a[i].lazy=0;
if(l==r)
{
a[i].sum=1;
return;
}
int mid=(l+r)>>1;
build(l,mid,i<<1);
build(mid+1,r,i<<1|1);
PushUp(i);
}

void update(int l,int r,int val,int i)
{
if(a[i].l>=l && a[i].r<=r)
{
a[i].lazy=val;
a[i].sum=(r-l+1)*val;
return;
}
PushDown(i);
int mid=(a[i].l+a[i].r)>>1;
if(mid>=r)update(l,r,val,i<<1);
else if(mid<l)update(l,r,val,i<<1|1);
else
{
update(l,mid,val,i<<1);
update(mid+1,r,val,i<<1|1);
}
PushUp(i);
}

int main()
{
int T=in();
for(int ii=1;ii<=T;ii++)
{
int n=in();
build(1,n,1);
int m=in();
while(m--)
{
int t1=in(),t2=in(),t3=in();
update(t1,t2,t3,1);
}
printf("Case %d: The total value of the hook is %d.\n",ii,a[1].sum);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: