持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第26天，点击查看活动详情

题目

我们定义，在以下情况时，单词的大写用法是正确的：

• 全部字母都是大写，比如 "USA"。
• 单词中所有字母都不是大写，比如 "leetcode"。
• 如果单词不只含有一个字母，只有首字母大写， 比如 "Google"。

给你一个字符串 word。如果大写用法正确，返回 true；否则，返回 false。

示例 1：

• 输入： word = "USA"
• 输出： true

示例 2：

• 输入： word = "FlaG"
• 输出： false

方法一：根据题目要求实现

思路及解法

根据题目要求，若单词的大写用法正确，则需要满足：

• 若第 1 个字母为大写，则其他字母必须均为大写或均为小写，即其他字母必须与第 2 个字母的大小写相同；

• 若第 1 个字母为小写，则其他字母必须均为小写。

根据以上规则，可以整理得到以下更简单的判断规则：

• 无论第 1 个字母是否大写，其他字母必须与第 2 个字母的大小写相同；

• 若第 1 个字母为小写，则需额外判断第 2 个字母是否为小写。

代码

class Solution {
    func detectCapitalUse(_ word: String) -> Bool {
        let words: [Character] = Array(word)
        if words.count >= 2 && words[0].isLowercase && words[1].isUppercase {
            return false
        }
        
        for i in 2..<words.count {
            if words[0].isLowercase != words[i].isLowercase {
                return false
            }
        }
        return true
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 为字符串的长度。我们需要遍历字符串中的每个字符。

• 空间复杂度：$O(n)$，其中 $n$ 为字符串的长度。因为 $swift$ 中需要用额外的空间 $words$ 来存储每个字符。

方法二：遍历

思路及解法

用 firstCh 记录下第一个字符， 遍历字符串，用 `` 记录下所有大写字符的数量，根据题目规则可以得出下面结论

• 如果 firstCh 为大写且 `` 等于字符串长度，则单词的大写用法是正确的。
• 如果 firstCh 为大写且 `` 等于 1，则单词的大写用法是正确的。
• 如果 firstCh 为小写且 `` 等于 0，则单词的大写用法是正确的。

代码

class Solution {
    func detectCapitalUse(_ word: String) -> Bool {
        var firstCh: Character = " "
        var uppercaseCount: Int = 0
        for ch in word {
            if firstCh == " " {
                firstCh = ch
            }
            if ch.isUppercase {
                uppercaseCount += 1
            }
        }
        
        return (firstCh.isUppercase && uppercaseCount == word.count) || (firstCh.isUppercase == true && uppercaseCount == 1) || (firstCh.isLowercase && uppercaseCount == 0)
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 为字符串的长度。我们需要遍历字符串中的每个字符。

• 空间复杂度：$O(1)$。