leetcode 520. Detect Capital 检查大小写 + 统计大写字母即可

骑猪看日落 2022-06-03 03:13 223阅读 0赞

Given a word, you need to judge whether the usage of capitals in it is right or not.

We define the usage of capitals in a word to be right when one of the following cases holds:

All letters in this word are capitals, like “USA”.  
All letters in this word are not capitals, like “leetcode”.  
Only the first letter in this word is capital if it has more than one letter, like “Google”.  
Otherwise, we define that this word doesn’t use capitals in a right way.  
Example 1:  
Input: “USA”  
Output: True  
Example 2:  
Input: “FlaG”  
Output: False

这道题给了我们一个单词，让我们检测大写格式是否正确，规定了三种正确方式，要么都是大写或小写，要么首字母大写，其他情况都不正确。那么我们要做的就是统计出单词中所有大写字母的个数cnt，再来判断是否属于这三种情况，如果cnt为0，说明都是小写，正确；如果cnt和单词长度相等，说明都是大写，正确；如果cnt为1，且首字母为大写，正确，其他情况均返回false，参见代码如下：

代码如下：

#include <iostream>
    #include <vector>
    #include <map>
    #include <set>
    #include <queue>
    #include <stack>
    #include <string>
    #include <climits>
    #include <algorithm>
    #include <sstream>
    #include <functional>
    #include <bitset>
    #include <numeric>
    #include <cmath>
    
    using namespace std;
    
    class Solution 
    {
    public:
        bool detectCapitalUse(string word) 
        {
            int cnt = 0, n = word.size();
            for (int i = 0; i < n; ++i) 
            {
                if (word[i] <= 'Z')
                    ++cnt;
            }
            return cnt == 0 || cnt == n || (cnt == 1 && word[0] <= 'Z');
        }
    };