Leetcode-520: 检测大写字母

题目链接

https://leetcode-cn.com/problems/detect-capital/

题目

给定一个单词，你需要判断单词的大写使用是否正确。

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

全部字母都是大写，比如"USA"。
单词中所有字母都不是大写，比如"leetcode"。
如果单词不只含有一个字母，只有首字母大写， 比如 "Google"。
否则，我们定义这个单词没有正确使用大写字母。

示例

示例 1:

输入: "USA"
输出: True

示例 2:

输入: "FlaG"
输出: False

注意: 输入是由大写和小写拉丁字母组成的非空单词。

思路一

直接根据题意来判断。

C++ Code

class Solution {
public:
    bool capital(char a){
        if(a-'A'>=0&&a-'A'<=25) return true;
        else return false;
    }
    bool detectCapitalUse(string word) {
        if( capital(word[0])== 1)//首字母是大写，后面的字母全都要是大写或者小写
        {
            //判断第二个字母
            if(capital(word[1])== 1) //第二个字母也大写 后面全部大写
            {
                int i=2;
                while(i<word.size())
                {
                    if(capital(word[i])==0) return false;
                    i++;
                }
                return true;
            }
            else //第二个字母小写，后面的全部要小写
            {
                int i=2;
                while(i<word.size())
                {
                    if(capital(word[i])==1) return false;
                    i++;
                }
                return true;
            }
        }
        else //首字母小写，后面的全部小写
        {
            int i=1;
            while(i<word.size())
            {
                if(capital(word[i])==1) return false;
                i++;
            }
            return true;
        }
    }
};

结果

以上代码貌似有些繁琐，可以进一步优化。

思路二

统计大小写字母的个数，三种情况一起判断。

C++ Code

class Solution {
public:
    bool isAA(char a){
        if(a>='A'&&a<='Z') return true;
        else return false;
    }
    bool detectCapitalUse(string word) 
    {
        int AA=0,aa=0,n=word.size(); //AA表示大写字母个数 aa表示小写字母个数
        if(n==1) return true;
        for(char ch:word)  isAA(ch)?AA++:aa++; //统计大小写个数
        //全为大写 或者 全为小写 或者 有且仅有首字母大写且其余均为小写
        if(AA==n||aa==n || (AA==1 && isAA(word[0]) && aa==n-1)) return true;
        return false;
    }
};

 结果