HDU 1696 Just a Hook
题目大意:
存在一串长度为n的序列,初始值全为1,给出q次操作,每次操作输入l、r、s,表示将区间 l 到 r 之间的值全部改为 s,其中1<=s<=3,序列长度1<=n<=100,000,操作次数0<=q<=100,000。要求输出q次操作之后序列的和。有多组输入样例。
大致思路:
因为是更改区间的值,所以会和区间上加减一个值有区别,对于这类题目我们用线段树进行处理。线段树的每一段表示这段序列区间内所有的值的大小,对于每次操作,都要对产生影响的区间内的值进行维护。
对于一段区间中既有未更新的又有更新的情况,将未更新的区间用父节点的值进行覆盖,同时将父节点清零,更新的区间用新值取代。
这里的区间都用[ l , r )左闭右开表示。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
static const int maxn = 270000;
int ST[maxn];
int n,tn,T,q;
void update(int a,int b,int mat,int k,int l,int r){
if(a<=l&&r<=b){ //该区间包含在更新区间内 更新节点值
ST[k] = mat;
return;
}
else if(b<=l||r<=a){ //区间和更新区间没有交集 用父节点的值覆盖(如果存在的话)
if(ST[k/2]!=0) ST[k] = ST[k/2];
return;
}
else{
if(ST[k/2]!=0) ST[k] = ST[k/2]; //二分子区间进行更新
update(a,b,mat,k*2,l,(l+r)/2);
update(a,b,mat,k*2+1,(l+r)/2,r);
ST[k] = 0;
return;
}
}
int getsum(int k,int l,int r){
if(ST[k]==0){
int mid = (l+r)/2;
if(mid>n) return getsum(k*2,l,mid);
else return getsum(k*2,l,mid)+getsum(k*2+1,mid,r);
}
else return ST[k]*(r-l);
}
int main()
{
scanf("%d",&T);
for(int t=1;t<=T;t++){
scanf("%d",&n);
tn = 1;
while(tn<n) tn<<=1;
fill(ST,ST+maxn,0);
update(1,n+1,1,1,1,tn+1);
scanf("%d",&q);
for(int i=1;i<=q;i++){
int l,r,material;
scanf("%d %d %d",&l,&r,&material);
update(l,r+1,material,1,1,tn+1);
}
printf("Case %d: The total value of the hook is %d.\n",t,getsum(1,1,tn+1));
}
return 0;
}
对于样例的每次update之后的线段树结构:
0
1 0
0 0 0 0
0 0 0 0 1 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
0
0 0
2 0 0 0
0 0 0 1 1 0 0 0
0 0 0 0 2 1 0 0 0 0 0 0 0 0 0 0
0
0 0
2 3 0 0
0 0 0 1 0 0 0 0
0 0 0 0 2 1 0 0 3 1 0 0 0 0 0 0