题目大意:给一个长度为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;
}