门|门 IT 2007 |第 55 题

题目描述

给出一个只包含字母的字符串，判断其中是否所有字母都相同，如果是则返回其长度，否则返回 0。

输入格式

一个只包含大小写字母的字符串。

输出格式

一个整数，表示字符串中所有字母相同的情况下的字符串长度，如果不是所有字母都相同则返回 0。

示例

输入：

aaaAAA

输出：

6

解题思路

循环遍历字符串，当发现当前字母和前一个字母不同，则说明不是所有字母都相同，直接返回 0，否则继续循环，直到遍历完整个字符串，返回字符串长度。

代码实现

def check_same(s: str) -> int:
    if len(s) < 2:
        return len(s)
    pre = s[0]
    for i in range(1, len(s)):
        if s[i] != pre:
            return 0
        pre = s[i]
    return len(s)

注意事项

• 本题不能忽略字母的大小写差异，需要全部考虑，例如 "aaaAAA" 不符合要求；
• 特判字符串长度小于 2 的情况，需要直接返回字符串长度；
• 循环遍历字符串时，需要注意遍历范围。