E. K-periodic Garland dp

cf原题链接

题目大意:给一个长度为n的01串,给定距离k,要求所有1之间的距离都为k

dp[i][0]:表示前i个数据合法且第i个数据为0
dp[i][1]:表示前i个数据合法且第i个数据为1

dp[i][0]=min(dp[i-1][0],dp[i-1][1])+(a[i]==‘1’)
因为dp[i][0]有两种情况,要么它是两个1之间(这种情况在计算dp[x][1]的时候会有判断,这里不用担心不符合要求),要么它后面一个1都没,无论哪种情况都可以直接从上一个的状态直接转移过来

dp[i][1]=min(sum[i-1],dp[i-k][1]+sum[i-1]-sum[i-k])+(a[i]==‘0’);
sum[i]表示前i个里有多少个1。dp[i][1]有两种情况,一种是它是第一个1,一种就是从上个1转移过来,(sum[i-1]-sum[i-k])是i和i-k两个位置之间的1的个数。

ac代码

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e6+5;
const int inf=1e9;
char aa[maxn];
int dp[maxn][2];
int sum[maxn];

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    int t;cin>>t;
    while(t--){
        int n,k;cin>>n>>k;
        sum[0]=0;dp[0][0]=0;dp[0][1]=0;
        for(int i=1;i<=n;i++){
            sum[i]=0;dp[i][1]=0;dp[i][1]=0;
            cin>>aa[i];
            if(aa[i]=='1') sum[i]=sum[i-1]+1;
            else sum[i]=sum[i-1];
        }
        for(int i=1;i<=n;i++){
            dp[i][0]=min(dp[i-1][0],dp[i-1][1])+(aa[i]=='1');
            if(i-k>=1) dp[i][1]=min(sum[i-1],dp[i-k][1]+sum[i-1]-sum[i-k])+(aa[i]=='0');
            else dp[i][1]=sum[i-1]+(aa[i]=='0');
        }
        cout<<min(dp[n][0],dp[n][1])<<endl;
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值