題意: 用01字符串表示長度爲n的長桌,'1’表示已經有人,'0’表示空位,每個人之間相隔至少k個空位是餐廳的規定。試問還有多少個可以選擇的空位,選擇後依舊符合規定。
思路:
- 模擬一下,如果第一個位置s[0]爲空就先選擇並記錄其位置,再判斷如果和後面的人衝突就放棄選擇。
- 同理處理後面的位置,從前往後數到k個空位,若第k+1爲空就先選擇並記錄其位置,若與後面的衝突就放棄選擇。
代碼實現:
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cctype>
#include <cstring>
#include <iostream>
#include <sstream>
#include <string>
#include <list>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <algorithm>
#include <functional>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll mod = 1e9 + 7;
const int N = 2e5 + 5;
int t, n, k;
signed main()
{
IOS;
cin >> t;
while(t --){
cin >> n >> k;
string s; cin >> s;
int pos = -1, cnt = 0, ans = 0;
for(int i = 0; i < n; i ++){
if(s[i] == '1'){
if(cnt < k && pos != -1){ //如果衝突就放棄選擇
s[pos] = '0';
ans --; //放棄選擇
}
cnt = 0;
}
else cnt ++;
if(!i && s[i] == '0'){ //如果第一個位置空就先選擇
pos = i; //記錄其位置
s[i] = '1';
ans ++;
cnt = 0;
}
if(cnt == k && i + 1 < n && s[i + 1] == '0'){ //如果有k個空位且第k+1個也是空位就先選擇
ans ++;
s[++ i] = '1';
pos = i; //記錄其位置
cnt = 0;
}
}
cout << ans << endl;
}
return 0;
}