LeetCode 520. Detect Capital

Description:

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, like “Google”.
Otherwise, we define that this word doesn’t use capitals in a right way.

Example 1:

Input: “USA”
Output: True

Solution

class Solution {
public:
    bool detectCapitalUse(string word) {
        int length = word.size();
    int flag = 0;
    if (length == 1)
        return true;
    if (word.at(0) >= 'a' && word.at(0) <= 'z') {
        flag = 1;
    } else if (word.at(0) >= 'A' && word.at(0) <= 'Z') {
        if (word.at(1) >= 'A' && word.at(1) <= 'Z')
            flag = 0;
        else if (word.at(1) >= 'a' && word.at(1) <= 'z')
            flag = 2;
        else
            return false;
    } else
        return false;

    if (length >= 2) {
        for (int i = 1; i < length; ++i) {
            if (flag == 0) {
                if (!(word.at(i) >= 'A' && word.at(i) <= 'Z'))
                    return false;
            } else if (flag == 1 || flag == 2) {
                if (!(word.at(i) >= 'a' && word.at(i) <= 'z'))
                    return false;
            }
        }
    }

    return true;
    }
};