Skip to main content

Codeforces Round #112 (Div. 2)(C. Another Problem on Strings)


 Bismillahir Rahmanir Rahim

///Author: Tanvir Ahmmad
///CSE,Islamic University,Bangladesh

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<sstream>
#include<cmath>
#include<cstring>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<stack>
#include<vector>
#include<iterator>
#include <functional> ///sort(arr,arr+n,greater<int>()) for decrement of array
/*every external angle sum=360 degree
  angle find using polygon hand(n) ((n-2)*180)/n*/
///Floor[Log(b)  N] + 1 = the number of digits when any number is represented in base b
using namespace std;
typedef long long ll;

int main()
{
    ll n,cnt=0,bro=0;
    map<ll,ll>mp;
    map<ll,ll>::iterator it;
    string str;
    cin>>n>>str;
    if(n)
    {
        ll sz=str.size(),arr[sz+3],k=0;
        for(ll i=0; i<sz; i++)
        {
            if(str[i]=='1')
                arr[k++]=i,mp[i]=bro,bro=0;
            else
                bro++;
        }
        arr[k++]=sz,mp[sz]=bro;
        for(ll i=0; i<k-n; i++)
        {
            ll pre=arr[i],post=arr[i+n];
            it=mp.find(pre);
            ll cnt_pre=it->second;
            it=mp.find(post);
            ll cnt_post=it->second;
            if(max(cnt_post,cnt_pre)==0)
                cnt++;
            else if(min(cnt_post,cnt_pre)==0 && max(cnt_post,cnt_pre)!=0)
                cnt+=(max(cnt_post,cnt_pre)+1);
            else
                cnt+=(1+cnt_pre+cnt_post+(cnt_pre*cnt_post));
        }
        cout<<cnt<<endl;
    }
    else
    {
        for(ll i=0;i<str.size();i++) 
        {
            if(str[i]=='0') bro++;
            else cnt+=(bro*(bro+1)/2),bro=0;
        }
        if(bro) cnt+=(bro*(bro+1)/2),bro=0;
        cout<<cnt<<endl;
    }
    return 0;
}
///Alhamdulillah

Problem Link:https://codeforces.com/problemset/problem/165/C
Labels:

Comments

Post a Comment

Popular posts from this blog

Codeforces round 1676(A. Lucky?)

Just count the  first three  &  last three  number ///Bismillahir Rahmanir Rahim ///Author: Tanvir Ahmmad ///CSE,Islamic University,Bangladesh #include< iostream > #include< cstdio > #include< algorithm > #include< string > #include< cstring > #include< sstream > #include< cmath > #include< cstring > #include< vector > #include< queue > #include< map > #include< set > #include< stack > #include< vector > #include< iterator > #include   < functional >   ///sort(arr,arr+n,greater<int>()) for decrement of array /*every external angle sum=360 degree   angle find using polygon hand(n) ((n-2)*180)/n*/ ///Floor[Log(b)  N] + 1 = the number of digits when any number is repres...
Post a Comment
Read more

Maximum Number of Coins You Can Get (LeetCode)

Problem Name: 1561.  Maximum Number of Coins You Can Get Problem Link: https://leetcode.com/problems/maximum-number-of-coins-you-can-get/description/ Difficulty: Medium Tag: Sorting | Array | Math | Greedy | Sorting | Game theory Language: C# | C++ OJ: LeetCode Algorithm: Sort the Piles: Sort the array of piles in descending order, so that the piles with the maximum number of coins are at the beginning. Calculate the Total Number of Piles to Collect From: Determine the total number of piles you can collect coins from based on the rule that for every three consecutive piles, you can only pick coins from two of them. Set this value to 2/3 of the total number of piles. Iterate Through Piles and Collect Coins: Start iterating through the sorted piles from the second pile (index 1) and continue up to the calculated total number of piles to collect from. In each iteration, collect the coins from the current pile. Since you can only collect coins from every second pile, use a step of ...
Post a Comment
Read more

Codeforces round 1661(B. Getting Zero)

using   second operation  the answer is maximum is  15  because  2 15 =32768 so we need to  pre-calculate all the illegible input 0 to 32768   using  second operation Then  just increment 1(using first operation)  from input check is it  minimum is not ! ///Bismillahir Rahmanir Rahim ///Author: Tanvir Ahmmad ///CSE,Islamic University,Bangladesh #include< iostream > #include< cstdio > #include< algorithm > #include< string > #include< cstring > #include< sstream > #include< cmath > #include< cstring > #include< vector > #include< queue > #include< map > #include< set > #include< stack > #include< vector > #include< iterator > #include   < functional >   ///sort(arr,arr+n,greater<int...
Post a Comment
Read more