F. Wi-Fi(线段树实现dp)

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

dp[i][0]:表示第i个位置不装的最小代价

dp[i][1]:表示第i个位置装的最小代价

T1的线段树是维护装的最小代价

T2的线段树是维护装和不装的最小代价

#include<bits/stdc++.h>
using namespace std;
#define lson root<<1,l,midd
#define rson root<<1|1,midd+1,r
typedef long long ll;
const int M=2e5+;
const ll INF=0x3f3f3f3f3f3f3f3f;
ll dp[M][];
char s[M];
struct node{
    ll tree[M<<];
    T(){memset(tree,INF,sizeof(tree));}
    void up(int root){
        tree[root]=min(tree[root<<],tree[root<<|]);
    }
    void update(int pos,ll val,int root,int l,int r){
        if(l==r){
            tree[root]=min(tree[root],val);
            return ;
        }
        int midd=(l+r)>>;
        if(pos<=midd)
            update(pos,val,lson);
        else
            update(pos,val,rson);
        up(root);
    }
    ll query(int L,int R,int root,int l,int r){
        if(L<=l&&r<=R){
            return tree[root];
        }
        int midd=(l+r)>>;
        ll ans=INF;
        if(L<=midd)
            ans=min(ans,query(L,R,lson));
        if(R>midd)
            ans=min(ans,query(L,R,rson));
        return ans;
    }
}T1,T2;
int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    scanf("%s",s+);
    memset(dp,INF,sizeof(dp));
    memset(T1.tree,INF,sizeof(T1.tree));
    memset(T2.tree,INF,sizeof(T2.tree));
    dp[][]=dp[][]=;
  //  cout<<dp[2][1]<<endl;
    T2.update(,,,,n);
    for(int i=;i<=n+;i++){
        dp[i][]=T1.query(max(,i-k),i-,,,n);
        if(s[i]=='')
            dp[i][]=T2.query(max(,i-k-),i-,,,n)+i-;
        else
            dp[i][]=min(min(dp[i-][],dp[i-][]),T1.query(max(,i-k-),i-,,,n))+i-;
        T2.update(i,min(dp[i][],dp[i][]),,,n);
        if(s[i]=='')
            T1.update(i,dp[i][],,,n);
    }
    printf("%I64d\n",min(dp[n+][],dp[n+][]));
    return ;
}