CodeForces - 635D Factory Repairs —— 点更新 区间查询

题意:

工厂生产产品,开始时每天生产b个,维修连续的k天后变成每天生产a个

第di天有需求量为ai的订单,每个订单只能当天生产,不能今天生产的明天卖

问若从某天开始维修,能完成的订单里的产品总数

思路:

用线段树维护,每个节点里分别记录这个区间内修理之前和修理之后能完成的产品总数,询问时分别查询修理前和修理后

要注意订单是可以叠加的,点更新时不是赋值是累加

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define ll long long
#define max_ 200100
#define mod 1000000007
#define inf 0x3f3f3f3f
struct node
{
    int l,r;
    ll w1,w2;
};
struct node tree[max_*4];
int n,k,q;
ll a,b;
void built(int i,int l,int r)
{
    tree[i].l=l;
    tree[i].r=r;
    tree[i].w1=0;
    tree[i].w2=0;
    if(l==r)
    return;
    int mid=(l+r)>>1;
    built(i<<1,l,mid);
    built(i<<1|1,mid+1,r);
}
void updata(int i,int x,ll v)
{
    if(tree[i].l==tree[i].r)
    {
        tree[i].w1=min(v+tree[i].w1,b);
        tree[i].w2=min(v+tree[i].w2,a);
        return;
    }
    int mid=(tree[i].l+tree[i].r)>>1;
    if(x<=mid)
    updata(i<<1,x,v);
    else
    updata(i<<1|1,x,v);
    tree[i].w1=tree[i<<1].w1+tree[i<<1|1].w1;
    tree[i].w2=tree[i<<1].w2+tree[i<<1|1].w2;
}
ll query(int i,int l,int r,int ti)
{
    if(tree[i].l==l&&tree[i].r==r)
    {
        if(ti==1)
        return tree[i].w1;
        else
        return tree[i].w2;
    }
    int mid=(tree[i].l+tree[i].r)>>1;
    if(r<=mid)
    return query(i<<1,l,r,ti);
    else if(l>mid)
    return query(i<<1|1,l,r,ti);
    else
    return query(i<<1,l,mid,ti)+query(i<<1|1,mid+1,r,ti);
}
int main(int argc, char const *argv[]) {
    scanf("%d%d%lld%lld%d",&n,&k,&a,&b,&q);
    built(1,1,n);
    while(q--)
    {
        int ti,data,num;
        scanf("%d",&ti);
        if(ti==1)
        {
            scanf("%d%d",&data,&num);
            updata(1,data,num);
        }
        else
        {
            scanf("%d",&data);
            ll ans=0;
            if(data>1)
            ans+=query(1,1,data-1,1);
            if(data+k<=n)
            ans+=query(1,data+k,n,2);
            printf("%lld\n",ans);
        }
    }
    return 0;
}

 

你可能感兴趣的:(Codeforce,线段树)