Codeforces1096B - Substring Removal

题目思路何在?

python3 代码

n, s = int(input()), input()
l = 0
while l < n:
    if s[l] != s[0]:
        break
    l += 1
r = n - 1
while r >= 0:
    if s[r] != s[-1]:
        break
    r -= 1
r = n - (r + 1)

ans = 0
if l == n:
    ans += (n * (n+1) // 2) % 998244353
else:
    if s[0] != s[-1]:
        ans += l + r + 1
    else:
        ans += ((l + 1) * (r + 1)) % 998244353
print(ans)