bzoj2119 [ZJOI2010]base基站选址

传送门

n年前的考试题，今天才填上……

听说你们会决策单调性+主席树？然而我多年不写决策单调性，懒得写了……于是就写了一发线段树。

其实线段树应该不难想，毕竟转移是分层转移，并且这个题的转移函数可以快速从$i$更新到$i+1$，这样就给线段树带来了不少方便。

每个村庄的$s$的限制可以看成一条线段，那么我们在扫过一个线段的右端点时把左端点以左的点都加上这条线段的代价，更新完之后求区间最小值即可，显然可以方便地用线段树来维护区间增量和区间最值。然后就没有然后了。

/**************************************************************
    Problem: 1835
    User: hzoier
    Language: C++
    Result: Accepted
    Time:6572 ms
    Memory:10592 kb
****************************************************************/
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=20010,maxk=110;
struct A{
    int l,r,w;
    bool operator<(const A &a)const{return r<a.r;}
}a[maxn];
void build(int,int,int);
void modify(int,int,int,int,int,int);
int query(int,int,int,int,int);
int mn[maxn<<2],lazy[maxn<<2];
int d[maxn],c[maxn],s[maxn],w[maxn];
int n,m,k,f[maxn][maxk],ans=0;
int main(){
    scanf("%d%d",&n,&m);
    for(int i=2;i<=n;i++)scanf("%d",&d[i]);
    for(int i=1;i<=n;i++)scanf("%d",&c[i]);
    for(int i=1;i<=n;i++)scanf("%d",&s[i]);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i].w);
        a[i].l=lower_bound(d+1,d+n+1,d[i]-s[i])-d;
        a[i].r=upper_bound(d+1,d+n+1,d[i]+s[i])-d-1;
        f[a[i].r+1][1]+=a[i].w;
        w[a[i].l-1]+=a[i].w;
        ans+=a[i].w;//printf("l=%d r=%d w=%d\n",a[i].l,a[i].r,a[i].w);
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)f[i][1]+=f[i-1][1];
    for(int i=n;i;i--)w[i]+=w[i+1];
    for(int i=1;i<=n;i++){
        f[i][1]+=c[i];
        ans=min(ans,f[i][1]+w[i]);
    }
    //for(int i=1;i<=n;i++)printf("f[%d][0]=%d w=%d\n",i,f[i][0],w[i]);
    f[1][1]=c[1];
    for(k=2;k<=m;k++){
        int cur=1;
        build(1,n,1);
        for(int i=k;i<=n;i++){
            while(cur<=n&&a[cur].r<i){
                if(a[cur].l>=k)modify(k-1,a[cur].l-1,a[cur].w,1,n,1);//printf("%d~%d += %d\n",1,a[cur].l-1,a[cur].w);
                cur++;
            }
            f[i][k]=query(k-1,i-1,1,n,1)+c[i];//printf("f[%d][%d]=%d c[i]=%d\n",i,k,f[i][k],c[i]);
            ans=min(ans,f[i][k]+w[i]);
        }
    }
    printf("%d",ans);
    return 0;
}
void build(int l,int r,int rt){
    lazy[rt]=0;
    if(l==r){
        mn[rt]=f[l][k-1];
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    mn[rt]=min(mn[rt<<1],mn[rt<<1|1]);
}
void modify(int s,int t,int d,int l,int r,int rt){
    if(s<=l&&t>=r){
        mn[rt]+=d;
        lazy[rt]+=d;
        return;
    }
    int mid=(l+r)>>1;
    if(s<=mid)modify(s,t,d,l,mid,rt<<1);
    if(t>mid)modify(s,t,d,mid+1,r,rt<<1|1);
    mn[rt]=min(mn[rt<<1],mn[rt<<1|1])+lazy[rt];
}
int query(int s,int t,int l,int r,int rt){
    if(s<=l&&t>=r)return mn[rt];
    int mid=(l+r)>>1;
    if(t<=mid)return query(s,t,l,mid,rt<<1)+lazy[rt];
    if(s>mid)return query(s,t,mid+1,r,rt<<1|1)+lazy[rt];
    return min(query(s,t,l,mid,rt<<1),query(s,t,mid+1,r,rt<<1|1))+lazy[rt];
}