[ZJOI2010] 基站选址

最新推荐文章于 2024-09-09 13:50:25 发布

weixin_30432007

最新推荐文章于 2024-09-09 13:50:25 发布

阅读量185

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/nosta/p/11032258.html

本文详细介绍了如何使用区间树状数组与动态规划解决复杂问题，通过具体实例讲解了算法的设计与实现过程，深入探讨了算法的优化技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

TheLostWeak ~~讲解得十分清楚，就不造轮子了~~

#include <bits/stdc++.h>
#define ll long long 
#define ls (x<<1)
#define rs (x<<1|1)
using namespace std;

const int N=2e4+10;
const int inf=0x3f3f3f3f;

int n,K;
int d[N],c[N],s[N],w[N],f[N];
vector<pair<int,int>> L[N];

int val[N<<2],tag[N<<2];
void psh(int x) {
    if(!tag[x]) return;
    val[ls]+=tag[x],tag[ls]+=tag[x];
    val[rs]+=tag[x],tag[rs]+=tag[x];
    tag[x]=0;
}
void build(int x,int l,int r) {
    tag[x]=0;
    if(l==r) {val[x]=f[l]; return;}
    int mid=(l+r)>>1;
    build(ls,l,mid); 
    build(rs,mid+1,r);
    val[x]=min(val[ls],val[rs]);
}
int query(int x,int l,int r,int L,int R) {
    if(L<=l&&r<=R) return val[x];
    int mid=(l+r)>>1,ret=inf; psh(x);
    if(L<=mid) ret=query(ls,l,mid,L,R);
    if(mid<R) ret=min(ret,query(rs,mid+1,r,L,R));
    return ret;
}
void modify(int x,int l,int r,int L,int R,int w) {
    if(L>R) return;
    if(L<=l&&r<=R) {val[x]+=w; tag[x]+=w; return;}
    int mid=(l+r)>>1; psh(x);
    if(L<=mid) modify(ls,l,mid,L,R,w);
    if(mid<R) modify(rs,mid+1,r,L,R,w);
    val[x]=min(val[ls],val[rs]);
}

int main() {
    scanf("%d%d",&n,&K); 
    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",w+i);
    n++; K++; d[n]=inf; w[n]=inf; 
    for(int i=1; i<=n; ++i) {
        int l=lower_bound(d+1,d+n+1,d[i]-s[i])-d;
        int r=upper_bound(d+1,d+n+1,d[i]+s[i])-d-1;
        L[r].push_back({l,w[i]});
    }
    for(int i=1,sum=0; i<=n; ++i) {
        f[i]=sum+c[i];
        for(auto p:L[i]) sum+=p.second;
    }
    for(int T=2; T<=K; ++T) {
        build(1,1,n); memset(f,inf,sizeof f);
        for(int i=T; i<=n; ++i) {
            for(auto p:L[i-1]) modify(1,1,n,T-1,p.first-1,p.second);
            f[i]=query(1,1,n,1,i)+c[i];
        }
    }
    printf("%d\n",f[n]);
    return 0;
}

转载于:https://www.cnblogs.com/nosta/p/11032258.html