Codeforces C. Social Distance (模拟) (Round #650 Div.3)

传送门

题意： 用01字符串表示长度为n的长桌，'1’表示已经有人，'0’表示空位，每个人之间相隔至少k个空位是餐厅的规定。试问还有多少个可以选择的空位，选择后依旧符合规定。

思路：

• 模拟一下，如果第一个位置s[0]为空就先选择并记录其位置，再判断如果和后面的人冲突就放弃选择。
• 同理处理后面的位置，从前往后数到k个空位，若第k+1为空就先选择并记录其位置，若与后面的冲突就放弃选择。

代码实现：

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair
#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;
}