HDU 1698 Just a Hook (线段树区间修改+区间查询)

题目链接: 传送门

题意:Pudge对装备钩子进行若干次的强化，强化分为三种分别对应的价值是1，2，3，在经历过若干次操作后，输出钩子对应的总价值，每次强化都是对钩子进行区间修改

解题思路:在明白了题意后，我们的线段树只需要维护一个区间的和值，然后每次操作的时候，lazy数组标记每次更改的区间，在查询的时候pushdown一下，注意的是在updata的时候，a[k]表示的是一个区间的值,所以是\((r-l+1)\times v\),最后记得多组输入(不然会Wa

Code:

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 400400;
int a[maxn],lazy[maxn];
void pushdown(int k,int l,int r)
{
    if(lazy[k])
    {
        lazy[k<<1]=lazy[k];
        lazy[k<<1|1]=lazy[k];
        int mid=(l+r)>>1;
        a[k<<1]=(mid-l+1)*lazy[k];
        a[k<<1|1]=(r-mid)*lazy[k];
        lazy[k]=0;
    }
}
void build(int k,int l,int r)
{
    lazy[k]=0;
    if(l==r)
    {
        a[k]=1;
        return ;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    a[k]=a[k<<1]+a[k<<1|1];
}
void update(int k,int l,int r,int x,int y,int v)
{
    if(x<=l&&y>=r)
    {
        lazy[k]=v;
        a[k]=(r-l+1)*v;
        return;
    }
    pushdown(k,l,r);
    int mid=(l+r)>>1;
    if(x<=mid) update(k<<1,l,mid,x,y,v);
    if(y>mid) update(k<<1|1,mid+1,r,x,y,v);
    a[k]=a[k<<1]+a[k<<1|1];
}
int query(int k,int l,int r,int x,int y)
{
    if(x<=l&&y>=r) return a[k];
    pushdown(k,l,r);
    int mid=(l+r)>>1;
    int ans=0;
    if(x<=mid) ans+=query(k<<1,l,mid,x,y);
    if(y>mid) ans+=query(k<<1|1,mid+1,r,x,y);
    return ans;
}
int main(void)
{
    int n,m,x,y,k,t,i,pt=1;
    scanf("%d",&t);
    while(t--)
    {
    	memset(lazy,0,sizeof lazy);
        scanf("%d%d",&n,&m);
        build(1,1,n);
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&x,&y,&k);
            update(1,1,n,x,y,k);
        }
        printf("Case %d: The total value of the hook is %d.\n",pt++,query(1,1,n,1,n));
    }
    return 0;
}