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

傳送門

題意: 用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;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章