【题解】StarryCoding P235 再短一点

题目传送门：P235 再短一点 | StarryCoding算法竞赛平台

题目描述

当一个长度为$m$的二进制字符串$T$是幸运的，我们可以通过任意顺序执行以下两种操作$m-1$来获得长度为$1$的字符串:

• 选择$T$中任意等于01的子字符串，将其替换为1。

• 选择$T$中任何等于10的子字符串，然后将其替换为0。

例如，如果$T=001$，我们可以选择子字符串$[T_2{T}_3]$并执行第一个操作。得到$T=01$。

您将得到一个长度为$n$的二进制字符串$S$，索引从$1$到$n$。找出$(l,r)$($1\le l\le r\le n$)使得$S[l\dots r]$($S$从$l$到$r$的子字符串)是一个幸运的字符串的整数对的个数。

输入描述

第一行包含一个整数$t$($1\le t\le 1000$)—测试用例的数量。下面是测试用例的描述。

每个测试用例的第一行包含一个整数$n$($1\le n\le 2\cdot 10^5$)——大小为$S$。

每个测试用例的第二行包含一个由$n$字符$S_1{S}_2\dots S_n$组成的二进制字符串$S$。(每个$1\le i\le n$对应$S_i=$ 0或$S_i=$ 1)

保证所有测试用例的$n$之和不超过$2\cdot 10^5$。

输出描述

对于每个测试用例，输出使$S[l\dots r]$($S$从$l$到$r$的子字符串)为偏执字符串的整数对$(l,r)1\le l\le r\le n$的个数。

输入样例1

5
1
1
2
01
3
100
4
1001
5
11111

输出样例1

1
3
4
8
5

样例解释

在第一个示例中，$S$已经具有长度$1$，不需要任何操作。

在第二个示例中，$S$的所有子字符串都是幸运的。对于整个字符串，执行第一个操作就足够了。

在第三个示例中，除了$[S_2{S}_3]$之外，$S$的所有子字符串都是幸运的，因为我们不能对它和$[S_1{S}_2{S}_3]$(整个字符串)执行任何操作。

思路

长度为$m$的二进制字符串$T$是幸运的，当且仅当$m=1$或($1<m$和$S[m]\ne S[m-1]$)。

• 在$S[m-1]=S[m]$的情况下:我们永远不能删除最后两个字符，因为它们将始终保持相等。所以 $S$ 不是幸运的。
• 在$S[m-1]\ne S[m]$的情况下:如果$m=2$，我们可以通过一次操作达到我们的目标。否则，假设最后一个字符为0。现在最后三个字符不是010就是110。在第一种情况下，对$[S_{m-2},S_{m-1}]$执行操作，在第二种情况下，对 $[S_{m-1},S_m]$ 执行操作。然后最后两个字符将是10，我们可以在新的字符串上继续这个算法，直到我们得到$m=1$。

长度为$1$的幸运的子字符串的个数等于$n$。为了计算更长的子字符串的个数，我们可以将索引$2$中的$r$固定为$n$。如果$S[r]\ne S[r-1]$成立，我们应该在答案上加上$r-1$。

复杂性: $\mathcal{O}(n)$

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5 + 9;

void solve()
{
    int n; cin >> n;
    ll ans = n;
    string s; cin >> s;
    for(int i = 1; i < n; ++i)
    {
        if(s[i] != s[i - 1]) ans += i;
    }

    cout << ans << '\n';
}

int main()
{
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int _; cin >> _;
    while(_ --)solve();
	return 0;
}

本题由codeforces上的1694B改编而成