Advertisement
Guest User

Untitled

a guest
Nov 6th, 2022
2,779
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.60 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. #define int long long
  3. using namespace std;
  4. int32_t main()
  5. {
  6.     cin.tie(nullptr)->ios_base::sync_with_stdio(false);
  7.     int q;
  8.     cin >> q;
  9.     while (q--)
  10.     {
  11.         int n;
  12.         string s;
  13.         cin >> n >> s;
  14.         s = '$' + s;
  15.         int cnt0 = 0, cnt1 = 0;
  16.         for (int i = 1; i <= n; ++i)
  17.         {
  18.             cnt0 += s[i] == '0';
  19.             cnt1 += s[i] == '1';
  20.         }
  21.         int ans = cnt0 * cnt1;
  22.         int lg = 1;
  23.         for (int i = 2; i <= n; ++i)
  24.         {
  25.             if (s[i] == s[i - 1])
  26.             {
  27.                 lg++;
  28.             }
  29.             else
  30.             {
  31.                 ans = max(ans, lg * lg);
  32.                 lg = 1;
  33.             }
  34.         }
  35.         ans = max(ans, lg * lg);
  36.         cout << ans << '\n';
  37.     }
  38. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement