Submission #71365950

---

Source Code Expand

#include <bits/stdc++.h>

using namespace std;

#ifdef LOCAL
#define debug(arg) cout << "[" << #arg << "]: " << arg << endl
#else
#define debug(arg) 42
#endif

using llu = unsigned long long;
using ll = long long;

#define vec vector
#define all(n) begin(n), end(n)
#define pb push_back

void solv() {
	int n; cin >> n;
	string s; cin >> s;
	vec<llu> R(n); for (auto &i : R) cin >> i;
	s += 'x';
	R.pb(0);
	llu mx = 0;
	llu mxc = 0;
	vec<tuple<llu, llu, bool>> groups;
	auto update = [&](llu cur) {
		if (cur > mx) mx = cur, mxc = 1;
		else if (cur == mx) ++mxc;
	};
	for (int l = 0, r = 0; r <= n; update(R[r++])) if (s[l] != s[r]) {
		groups.pb({mx, mxc, s[l] != '#'});
		mx = 0, mxc = 0;
		l = r;
	}
	if (!get<2>(groups.front())) groups.erase(groups.begin());
	if (!get<2>(groups.back())) groups.erase(--groups.end());
	auto it = max_element(all(groups));
	mx = get<0>(*it);
	llu ans = 0;
	for (int i = 2; i < groups.size(); i += 2) {
		auto &[l, lc, _1] = groups[i - 2];
		auto &[m, _2, _3] = groups[i - 1];
		auto &[r, rc, _4] = groups[i - 0];
		debug(l << ' ' << lc);
		debug(m);
		debug(r << ' ' << rc);
		if (l == mx || m == mx || r == mx) ans += lc*rc;
	}
	cout << ans << '\n';
}

signed main() {
	ios_base::sync_with_stdio(0), cin.tie(0);
	int t = 1;
	//cin >> t;
	while (t--) solv();
	return 0;
}

Submission Info

Compile Error

./Main.cpp:41:20: warning: comparison of integers of different signs: 'int' and 'size_type' (aka 'unsigned long') [-Wsign-compare]
   41 |         for (int i = 2; i < groups.size(); i += 2) {
      |                         ~ ^ ~~~~~~~~~~~~~
1 warning generated.

Judge Result