-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path2026C.cpp
More file actions
82 lines (69 loc) · 1.45 KB
/
2026C.cpp
File metadata and controls
82 lines (69 loc) · 1.45 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// coded by Cocane
#include <bits/stdc++.h>
#define ll long long
#define forn(i, n) for (int i = 0; i < n; i++)
#define for1(i, n) for (int i = 1; i <= n; i++)
#define foru(i, n) for (int i = n - 1; i >= 0; i--)
#define fi first
#define se second
#define all(x) (x).begin(), (x).end()
#define allp(x) (x).begin(), (x).begin() + (x).size() / 2, (x).rbegin()
#define pb push_back
#define ppb pop_back
#define read(a) \
for (auto &i : a) \
cin >> i
#define out(ans) cout << ans << endl
#define yn(a) cout << (a ? "YES" : "NO") << endl
using namespace std;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef double dl;
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);
int t;
cin >> t;
while (t--)
{
int n;
cin >> n;
string s;
cin >> s;
vi pre0, pre1;
for (int i = 0; i < n - 1; i++)
{
if (s[i] == '0')
pre0.pb(i);
else
pre1.pb(i);
}
reverse(all(pre1)); // Preserving reverse logic as is
int pi = pre0.size() - 1;
int pola = pre1.size() - 1;
int mal = n * (n + 1) / 2;
for (int i = n - 1; i >= 0; i--)
{
while (pi >= 0 && pre0[pi] >= i)
pi--;
while (pola >= 0 && pre1[pola] >= i)
pola--;
if (s[i] == '1')
{
if (pi >= 0 && pre0[pi] < i)
{
mal -= (i + 1);
pi--;
}
else if (pola >= 0 && pre1[pola] < i)
{
mal -= (i + 1);
pola--;
}
}
}
out(mal);
}
return 0;
}