Codeforces Round #587 (Div. 3) F. Wi-Fi dp+线段树维护

题目链接： http://codeforces.com/contest/1216/problem/F

题意：

$n$ 户人家要联网，第 $i$ 家用户联网的成本为 $i$ ，其中有一些人家家中有 $wifi$ ，这使得如果这些人家里联了网，那么 $[max(i-k,1),min(i+k,n)]$ 范围内的人家都可以连上网，问你所有人都连上网的最小成本。

做法：

我们假设某一户人家不直接联网，那么他一定通过某个有 $wifi$ 的人家间接联网，但是第 $i$ 户人家的联网成本为 $i$ ，所以如果一定要让他间接联网，那么为了使成本小，他一定会选择前面的 $wifi$ 人家，即转移是没有后效性的，符合 $dp$ 的性质。

我们设 $dp[i][0/1]$ 为到第 $i$ 户人家 不连/连 网（直接联网）的最小成本。

• 如果这户人家不联网，那么选取前面某个 $k$ 距离范围内有联网有 $wifi$ 且成本最小的人家作为自己的成本。
• 如果这户人家联网：

① 如果这户人家没有 $wifi$ ，那么他会从前一户人家的 $[0/1]$ 里面选取一个最小值+安装费用。
② 如果这户人家有 $wifi$，他会从：

1. 前面 $[i-2k-1,i-1]$ 有 $wifi$ 并且直接联网。
2. 前面 $[i-k-1,i-1]$ 没有 $wifi$并且直接联网。

上面两种情况中选取一个最小值+安装费用。

上面的区间最小值用线段树维护即可。

代码

#include<bits/stdc++.h>
#define lson rt<<1
#define rson rt<<1|1
#define mid (l+r)/2
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=200005;
const ll inf=1e15;
ll mi[maxn<<2][2],dp[maxn][2];
int n,k,a[maxn];

void build(int l,int r,int rt){
    mi[rt][0]=mi[rt][1]=inf;
    if(l==r) return ;
    build(l,mid,lson);
    build(mid+1,r,rson);
}
void update(int l,int r,int rt,int pos,ll v,int p){
    if(l==r){
        mi[rt][p]=v;
        return ;
    }
    if(pos<=mid) update(l,mid,lson,pos,v,p);
    else update(mid+1,r,rson,pos,v,p);
    mi[rt][p]=min(mi[lson][p],mi[rson][p]);
}
ll query(int l,int r,int rt,int ql,int qr,int p){
    if(ql<=l&&r<=qr){
        return mi[rt][p];
    }
    ll ls=inf,rs=inf;
    if(ql<=mid) ls=query(l,mid,lson,ql,qr,p);
    if(qr>mid) rs=query(mid+1,r,rson,ql,qr,p);
    return min(ls,rs);
}
int main(){
    scanf("%d%d",&n,&k);
    build(1,n,1);
    rep(i,1,n) {
        dp[i][0]=dp[i][1]=inf;
        scanf("%1d",&a[i]);
    }

    dp[1][1]=1;
    if(a[1]==1) update(1,n,1,1,1,0);
    else update(1,n,1,1,1,1);
    for(int i=2;i<=n;i++){
        int ql=max(1,i-k);
        ll Qmi=query(1,n,1,ql,i-1,0);
        if(Qmi!=inf) dp[i][0]=Qmi;
        ll tans=min(dp[i-1][1],dp[i-1][0])+i;
        if(a[i]==1){
            if(i-k<=1) dp[i][1]=i;
            else{
                Qmi = min(query(1,n,1,max(i-2*k-1,1),i-1,0),query(1,n,1,i-k-1,i-1,1));
                if(Qmi!=inf){
                    dp[i][1]=min(tans,Qmi+i);
                }
                else {
                    dp[i][1]=tans;
                }
            }
            update(1,n,1,i,dp[i][1],0);
        }
        else{
            dp[i][1]=tans;
            update(1,n,1,i,dp[i][1],1);
        }
    }
    printf("%lld\n",min(dp[n][0],dp[n][1]));
    return 0;
}