您的位置:首页 > 其它

hdu1698 Just a Hook 线段树区间更新,模板题

2017-07-25 21:06 323 查看
给你T组数据,N个数(初始时每个数的值为1),M个操作,每个操作把区间[a,b]里的数更新为c,问最后这N个数的和是多少。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
int l,r,w,lazy;
}a[100000*4];
void up(int i)
{
a[i].w=a[i*2].w+a[i*2+1].w;
}
void build(int l,int r,int i)
{
a[i].l=l;
a[i].r=r;
if(l==r)
{
a[i].w=1;
a[i].lazy=1;
return;
}
a[i].lazy=0;
int mid=(l+r)/2;
build(l,mid,i*2);
build(mid+1,r,i*2+1);
up(i);
}
void down(int i)
{
if(a[i].l==a[i].r)
return;
a[i*2].lazy=a[i].lazy;
a[i*2+1].lazy=a[i].lazy;
a[i*2].w=a[i].lazy*(a[i*2].r-a[i*2].l+1);
a[i*2+1].w=a[i].lazy*(a[i*2+1].r-a[i*2+1].l+1);
a[i].lazy=0;
}
void update(int l,int r,int k,int i)
{
if(a[i].l==l&&a[i].r==r)
{
a[i].w=k*(r-l+1);
a[i].lazy=k;
}
else
{
if(a[i].lazy)
down(i);
int mid=(a[i].l+a[i].r)/2;
if(r<=mid)
update(l,r,k,i*2);
else if(l>mid)
update(l,r,k,i*2+1);
else
{
update(l,mid,k,i*2);
update(mid+1,r,k,i*2+1);
}
up(i);
}
}
int main(void)
{
int i,j,k,n,m,t,kk=1,x,y;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
build(1,n,1);
scanf("%d",&m);
for(i=0;i<m;i++)
{
scanf("%d%d%d",&x,&y,&k);
update(x,y,k,1);
}
printf("Case %d: The total value of the hook is %d.\n",kk++,a[1].w);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: