leetcode第 185 场周赛——A.重新格式化字符串 题解

题目链接:https://leetcode-cn.com/problems/reformat-the-string/

leetcode第 185 场周赛——A.重新格式化字符串 题解_第1张图片
leetcode第 185 场周赛——A.重新格式化字符串 题解_第2张图片
要求字母、数字轮流出现,维护两个vector存储每个字母和数字,并用cnt1和cnt2记录字母和数字出现的次数,如果某一种字符比另一种出现的次数多2及以上,就无法满足,直接return空字符串。

分情况讨论,依次从两个vector中pop出字符放到结果字符串中

代码如下:

class Solution {
public:
    string reformat(string s) {
        vector<char> letter;
        vector<char> str; 
        string res = "";
        int len = s.size();
        int cnt1 = 0, cnt2 = 0;
        for(int i = 0; i < len; i++) {
            if(s[i] <= 'z' && s[i] >= 'a') {
                cnt1++;
                letter.push_back(s[i]);
            } else {
                cnt2++;
                str.push_back(s[i]);
            }
        }
        if(cnt1 > cnt2 + 1 || cnt2 > cnt1 + 1) {
            return "";
        }
        if(cnt1 > cnt2) {
            for(int i = 0; i < cnt2; i++) {
                res += letter.back();
                letter.pop_back();
                res += str.back();
                str.pop_back();
            }
            res += letter.back();
            return res;
        } else if(cnt1 < cnt2) {
            for(int i = 0; i < cnt1; i++) {
                res += str.back();
                str.pop_back();
                res += letter.back();
                letter.pop_back();
            }
            res += str.back();
            return res;
        } else {
            for(int i = 0; i < cnt2; i++) {
                res += letter.back();
                letter.pop_back();
                res += str.back();
                str.pop_back();
            }
            return res;
        }
    }
};

你可能感兴趣的:(Leetcode)