BZOJ 1835 基站选址（线段树优化DP）

struct Node
{
    int L,R;
    i64 Min,det;
     
    void set(i64 x)
    {
        det+=x;
        Min+=x;
    }
};
 
Node a[N<<2];
 
void pushUp(int t)
{
    if(a[t].L==a[t].R) return;
    a[t].Min=min(a[t*2].Min,a[t*2+1].Min);
}
 
i64 f[N],ans;
 
void build(int t,int L,int R)
{
    a[t].L=L;
    a[t].R=R;
    a[t].det=0;
    if(L==R)
    {
        a[t].Min=f[L];
        return;
    }
    int mid=(L+R)>>1;
    build(t*2,L,mid);
    build(t*2+1,mid+1,R);
    pushUp(t);
}
 
void pushDown(int t)
{
    if(a[t].L==a[t].R) return;
    if(a[t].det)
    {
        a[t*2].set(a[t].det);
        a[t*2+1].set(a[t].det);
        a[t].det=0;
    }
}
 
 
void add(int t,int L,int R,i64 x)
{
    if(L>a[t].R||R<a[t].L) return;
 
    if(L<=a[t].L&&a[t].R<=R)
    {
        a[t].set(x);
        return;
    }
     
    pushDown(t);
    add(t*2,L,R,x);
    add(t*2+1,L,R,x);
    pushUp(t);
}
 
i64 query(int t,int L,int R)
{
    if(L>a[t].R||R<a[t].L) return inf;
    if(L<=a[t].L&&a[t].R<=R) return a[t].Min;
     
    pushDown(t);
    i64 ans=min(query(t*2,L,R),query(t*2+1,L,R));
    pushUp(t);
    return ans;
}
 
 
 
int n,m,d[N],c[N],s[N],w[N];
int st[N],ed[N];
vector<int> V[N]; 
 
 
 
int getL(int x,int pos)
{
    int low=1,high=pos,mid;
    while(low<=high)
    {
        mid=(low+high)>>1;
        if(d[mid]>=x) high=mid-1;
        else low=mid+1;
    }
    if(high>=1&&d[high]>=x) return high;
    return low;
}
 
int getR(int x,int pos)
{
    int low=pos,high=n,mid;
    while(low<=high)
    {
        mid=(low+high)>>1;
        if(d[mid]>x) high=mid-1;
        else low=mid+1;
    }
    if(low<=n&&d[low]<=x) return low;
    return high;
}
 
void init()
{
    int i;
    FOR1(i,n)
    {
        st[i]=getL(d[i]-s[i],i);
        ed[i]=getR(d[i]+s[i],i);
        V[ed[i]].pb(i);
    }
}
 
 
void DP()
{
    build(1,0,n);
    int i,j,k;
    FOR1(i,n)
    {
        f[i]=query(1,0,i-1)+c[i];
        FOR0(j,SZ(V[i]))
        {
            k=V[i][j];
            add(1,0,st[k]-1,w[k]);
        }
    }
    upMin(ans,f[n]);
}
 
int main()
{
    RD(n,m);
    int i;
    for(i=2;i<=n;i++) RD(d[i]);
    FOR1(i,n) RD(c[i]);
    FOR1(i,n) RD(s[i]);
    FOR1(i,n) RD(w[i]);
    FOR1(i,n) f[i]=inf; 
    init(); 
    ans=inf; n++; m++;
    FOR1(i,m) DP();
    PR(ans);
}