Codeforces Round #650 (Div. 3) C. Social Distance (思维)

题目传送
题意:
给你一个01字符串，长度为n，并且其中的字符串已经满足，任意俩个1之间的距离大于k，现在让你插入尽可能多的1，使得还满足这个条件。

思路:
友情提示，如果没睡好，千万不要打cf…不然就是凭感觉在a题了
分情况讨论即可:
1.当全部是0时:
我们先在第0个位置上安一个1，然后后面一直找000…01(k个0)这种子结构就可以了，所以答案为: 1 + (n-1)/(k+1) 找还能有多少个上述子结构。

2.当有1的时候:
(1)我们先看第0个位置能不能安一个1，如果能，回到1情况，只不过这个时候的就是 1 + (还有多少区间长度)/(k+1)

(2)然后考虑俩个1之间的情况，求出俩个1之间的长度(不包含边界)，那么由于右边界为1，那么我们还要让出来k个0出来，然后一样的操作 答案加上 (还有多少区间长度)/(k+1)

(3)然后再考虑最后一个1到字符串末尾的长度区间，答案加上(区间长度)/(k+1)

AC代码

#include <bits/stdc++.h>
inline long long read(){char c = getchar();long long x = 0,s = 1;
while(c < '0' || c > '9') {if(c == '-') s = -1;c = getchar();}
while(c >= '0' && c <= '9') {x = x*10 + c -'0';c = getchar();}
return x*s;}
using namespace std;
#define NewNode (TreeNode *)malloc(sizeof(TreeNode))
#define Mem(a,b) memset(a,b,sizeof(a))
#define lowbit(x) (x)&(-x)
const int N = 1e6 + 10;
const long long INFINF = 0x7f7f7f7f7f7f7f;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-7;
const long long mod = 1e9+7;
const double II = acos(-1);
const double PP = (II*1.0)/(180.00);
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> piil;
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t;
    cin >> t;
    while(t--)
    {
        int n,k;
        cin >> n >> k;
        string s;
        cin >> s;
        int arr[n+5] = {0},m = 0;
        for(int i = 0;i < n;i++)
            if(s[i] == '1')
                arr[m++] = i;
        if(m == 0) cout << 1 + (n-1)/(k+1) << endl;
        else
        {
            int sum = 0;
            if(arr[0] != 0 && arr[0] >= k+1) //(1)
                sum = sum + 1 + (arr[0] - k - 1)/(k+1);
            for(int i = 1;i < m;i++) //(2)
                sum = sum + (arr[i]-arr[i-1]-k-1)/(k+1);
            sum = sum + (n-1 - arr[m-1])/(k+1); //(3)
            cout << sum << endl;
        }
    }
}