fork download
copy 
  1. #include "bits/stdc++.h"
  2. using namespace std;
  3. typedef long long ll;
  4.  
  5. int main(){
  6.     ios::sync_with_stdio(false);
  7.     cin.tie(nullptr);
  8.     int t; 
  9.     cin >> t;
  10.     while(t--){
  11.         ll n;
  12.         queue<int> q;
  13.         string s;
  14.         cin >> n >> s;
  15.         ll time = n/2 - 1;
  16.         for(int i = 0; i < n; i++){
  17.             if(s[i] == '1'){
  18.                 if(i > 2 * q.size()){
  19.                     q.push(i+1);
  20.                 }
  21.                 else{
  22.                     q.pop();
  23.                     q.push(i+1);
  24.                 }
  25.             }
  26.         }
  27.         ll sum = n * (n+1) / 2;
  28.         while(!q.empty()){
  29.             sum -= q.front();
  30.             q.pop();
  31.         }
  32.         cout << sum << endl;
  33.     }
  34.     return 0;
  35. }
Success #stdin #stdout 0.01s 5288KB
comments (?)
stdin
copy 
4
1
1
6
101101
7
1110001
5
11111
stdout
copy 
1
8
18
6
https://ideone.com/zJqyn9
language:
C++ (gcc 8.3)
created:
215 days ago
visibility:
public

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!