LeetCode 520 检测大写字母[模拟] HERODING的LeetCode之路

解题思路

模拟题就按照题目意思来做，题目分三种情况，那我们也分三种情况，通过设置flag用来标志第一位是大写还是小写，小写的话，直接通过后面的都不能大写来判断，大写的话，用count记录后面的大写的个数，count为1或者为字符串长度才符合要求，代码如下：

代码

class Solution {
public:
    // 判断是否是小写
    bool judge(char c) {
        return (c <= 'z' && c>= 'a');
    }

    bool detectCapitalUse(string word) {
        // flag用来标志第一位是大写还是小写
        bool flag = judge(word[0]);
        int count = 1;
        int len = word.length();
        for(int i = 1; i < len; i ++) {
            if(flag) {
                // 第一位小写要求后面的都不能大写
                if(!judge(word[i])) {
                    return false;
                } 
            } else {
                if(!judge(word[i])) {
                    count ++;
                } 
            }
        }
        if(!flag) {
            // 第一位大写要求后面的要不全小写，要不全大写
            if(count == 1 || count == len) {
                return true;
            } else {
                return false;
            }
        }
        return true;
    }
};