码迷,mamicode.com
首页 > 其他好文 > 详细

Codeforces Round #650 (Div. 3) C. Social Distance (前缀和)

时间:2020-06-19 14:26:17      阅读:52      评论:0      收藏:0      [点我收藏+]

标签:pre   span   alt   图片   --   first   air   size   ons   

技术图片

  • 题意:有一排座位,要求每人之间隔\(k\)个座位坐,\(1\)代表已做,\(0\)代表空座,问最多能坐几人.

  • 题解:我们分别从前和从后跑个前缀和,将已经有人坐的周围的位置标记,然后遍历求每一段连续的\(0\),对于每一段最多能坐\(\lceil len/(k+1) \rceil\),求个和就可.

  • 代码:

    #include <iostream>
    #include <cstdio>
    #include <cstring>
    #include <cmath>
    #include <algorithm>
    #include <stack>
    #include <queue>
    #include <vector>
    #include <map>
    #include <set>
    #include <unordered_set>
    #include <unordered_map>
    #define ll long long
    #define fi first
    #define se second
    #define pb push_back
    #define me memset
    const int N = 1e6 + 10;
    const int mod = 1e9 + 7;
    const int INF = 0x3f3f3f3f;
    using namespace std;
    typedef pair<int,int> PII;
    typedef pair<ll,ll> PLL;
     
    int t;
    int n,k;
    string s;
    bool st[N];
    vector<int> v;
     
    int main() {
        ios::sync_with_stdio(false);cin.tie(0);
        cin>>t;
        while(t--){
            cin>>n>>k;
            cin>>s;
            int len=s.size();
            int cnt=0;
            v.clear();
            me(st,0,sizeof(st));
            for(int i=0;i<len;++i){
                if(!st[i] && cnt>0){
                    st[i]=true;
                }
                cnt--;
                if(s[i]==‘1‘){
                    cnt=k;
                }
            }
            cnt=0;
            for(int i=len-1;i>=0;--i){
                if(!st[i] && cnt>0){
                    st[i]=true;
                }
                cnt--;
                if(s[i]==‘1‘){
                    cnt=k;
                }
            }
            int pos=-1;
            for(int i=0;i<len;++i){
                if(s[i]==‘1‘ || st[i]){
                    if(i-pos-1>0) v.pb(i-pos-1);
                    pos=i;
                }
            }
            if(s[len-1]==‘0‘ && !st[len-1]){
                if(len-1-pos>0) v.pb(len-1-pos);
            }
            int ans=0;
            for(auto w:v){
                ans+=(w-1)/(k+1)+1;
            }
            cout<<ans<<endl;
        }
        return 0;
    }
    

Codeforces Round #650 (Div. 3) C. Social Distance (前缀和)

标签:pre   span   alt   图片   --   first   air   size   ons   

原文地址:https://www.cnblogs.com/lr599909928/p/13162350.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!