题目链接:https://leetcode-cn.com/problems/reformat-the-string/
要求字母、数字轮流出现,维护两个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;
}
}
};