Ankit_132

B

May 29th, 2024
813
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.54 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. int main() {
  5.     int t;
  6.     cin>>t;
  7.     while(t--)
  8.     {
  9.         int n;
  10.         string s;
  11.         cin >> n >> s;
  12.        
  13.         int val = 0;
  14.         map<int, int> sum;
  15.         sum[0]++;
  16.        
  17.         long long int ans = n *1ll* (n + 1) / 2;
  18.        
  19.         for (auto e : s){
  20.             if (e == '1')
  21.                 val++;
  22.             else
  23.                 val--;
  24.    
  25.             sum[val]++;
  26.             ans += sum[val];
  27.         }
  28.    
  29.         cout << ans << "\n";
  30.     }
  31. }
Advertisement
Add Comment
Please, Sign In to add comment