String especial

Uma string é considerada "especial" se uma de duas condições for atendida:

• Todos os caracteres são iguais (por exemplo, "aaa").
• Todos os caracteres, exceto o do meio, são iguais (por exemplo, "aadaa").

Uma substring especial é qualquer substring de uma string que atende a um desses critérios. Dada uma string, determine quantas substrings especiais podem ser formadas a partir dela.

Dada a string s = "mnonopoo", temos as seguintes substrings especiais:

{ "m", "n", "o", "n", "o", "p", "o", "o", "oo", "non", "ono", "opo" }

Você só precisa retornar o número total dessas substrings especiais. Neste caso, é 12. Outro exemplo é s = "aaaa", então as substrings serão:

{ "a", "a", "a", "a", "aa", "aa", "aa", "aaa", "aaa", "aaaa" }
# The function will return 10

Exemplos

special_string("mnonopoo") ➞ 12

special_string("asasd") ➞ 7

special_string("aaaa") ➞ 10

Notas

N/A