题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1698
线段树典型的染色操作
My code:
//STATUS:C++_AC_359MS_3328KB
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
#define LL __int64
const int MAX=100010,INF=200000000;
int min(int x,int y){return x<y?x:y;}
struct Tree{
int l,r,num;
}tree[MAX*4];
void build(int l,int r,int k);
void update(int l,int r,int k,int z);
void search(int l,int r,int k);
int T,n,m,ans;
int main()
{
// freopen("in.txt","r",stdin);
int i,x,y,z,k=1;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
build(1,n,1);
for(i=0;i<m;i++){
scanf("%d%d%d",&x,&y,&z);
update(x,y,1,z);
}
ans=0;
search(1,n,1);
printf("Case %d: The total value of the hook is %d.\n",k++,ans);
}
return 0;
}
void build(int l,int r,int k)
{
tree[k].l=l,tree[k].r=r;
tree[k].num=1;
if(l==r)return;
int mid=(l+r)>>1;
build(l,mid,k<<1);
build(mid+1,r,(k<<1)+1);
}
void update(int l,int r,int k,int z)
{
if(tree[k].l==l && tree[k].r==r || tree[k].num==z){
tree[k].num=z;
return;
}
if(tree[k].num!=z && tree[k].num){
tree[k<<1].num=tree[k].num;
tree[(k<<1)+1].num=tree[k].num;
tree[k].num=0;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid)
update(l,r,k<<1,z);
else if(l>mid)
update(l,r,(k<<1)+1,z);
else {
update(l,mid,k<<1,z);
update(mid+1,r,(k<<1)+1,z);
}
}
void search(int l,int r,int k)
{
if(tree[k].l==l && tree[k].r==r && tree[k].num){
ans+=tree[k].num*(r-l+1);
return;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid)
search(l,r,k<<1);
else if(l>mid)
search(l,r,(k<<1)+1);
else {
search(l,mid,k<<1);
search(mid+1,r,(k<<1)+1);
}
}