hdu 1968 Just a Hook 线段树区间更新

线段树区间更新

题意：胖子有一条大金属棒，大金属棒由n个小金属棒组成，每次操作将一个区间的小金属棒变成金银铜三者之一，最后取出所有区间的金属棒总价值，注意，金属棒的初始值1

思路：每次把该区间内的值更新为该值，并更新父节点

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<algorithm>
#define maxn 100100
using namespace std;
int sto[maxn<<2];
int up[maxn<<2];
void Insert(int left,int right,int o)
{
up[o]=0;
if(left==right)
{
sto[o]=1;
return ;
}
int lso=o<<1;
int mid=(left+right)>>1;
Insert(left,mid,lso);
Insert(mid+1,right,lso|1);
sto[o]=sto[lso]+sto[lso|1];
}
void changedata(int num,int o)//更新节点
{
if(up[o]!=0)
{
int lso=o<<1;
up[lso]=up[o];
up[lso|1]=up[o];
sto[lso]=(num-(num>>1))*up[o];
sto[lso|1]=(num>>1)*up[o];
up[o]=0;
}
}
int getdata(int left,int right,int o,int begin,int end,int val)
{
if(begin<=left&&right<=end)//找到区间更新区间
{
up[o]=val;
sto[o]=(right-left+1)*val;
return sto[o];
}
changedata(right-left+1,o);
int lso=o<<1;
int mid=(left+right)>>1;
int ans=0;
if(begin<=mid)ans+=getdata(left,mid,lso,begin,end,val);
if(mid<end)ans+=getdata(mid+1,right,lso|1,begin,end,val);
sto[o]=sto[lso]+sto[lso|1];
return ans;
}
int main()
{
int t,n,m,ans=0;
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
Insert(1,n,1);//初始值全为1
int x,y,z;
scanf("%d",&m);
while(m--)
{
scanf("%d%d%d",&x,&y,&z);
int k=getdata(1,n,1,x,y,z);
}
printf("Case %d: The total value of the hook is %d.\n",++ans,sto[1]);
}
return 0;
}