题目描述:
给你一个字符串 s ,仅反转字符串中的所有元音字母,并返回结果字符串。
元音字母包括 ‘a’、‘e’、‘i’、‘o’、‘u’,且可能以大小写两种形式出现。
示例1:
输入:s = “hello”
输出:“holle”
示例2:
输入:s = “leetcode”
输出:“leotcede”
提示:
1 <= s.length <= 3 * 105
s 由 可打印的 ASCII 字符组成
代码:
class Solution {
public:
string reverseVowels(string s) {
int n=s.length(),a=-1,b=n,flag_a=0,flag_b=0;
if(n==1)return s;
while(a<b&&b>=0&&a<n){
// cout<<"a="<<a<<" flag_a="<<flag_a<<" b="<<b<<" flag_b="<<flag_b<<endl;
if(flag_a==0)a++;
if(flag_b==0)b--;
if(flag_a==1&&flag_b==1){
// cout<<"交换"<<a<<" "<<b<<" s["<<a<<"]="<<s[a]<<" s["<<b<<"]="<<s[b]<<endl;
swap(s[a],s[b]);
a++;
b--;
flag_a=0;
flag_b=0;
}
if(s[a]=='a'||s[a]=='A'||s[a]=='e'||s[a]=='E'||s[a]=='i'||s[a]=='I'||s[a]=='o'||s[a]=='O'||s[a]=='u'||s[a]=='U')
flag_a=1;
else
flag_a=0;
if(s[b]=='a'||s[b]=='A'||s[b]=='e'||s[b]=='E'||s[b]=='i'||s[b]=='I'||s[b]=='o'||s[b]=='O'||s[b]=='u'||s[b]=='U')
flag_b=1;
else
flag_b=0;
}
return s;
}
};
设置前后两个下标(指针)a和b,a从前向后遍历,b从后向前遍历,当a遇到元音字母,a停下等待b也遇到元音字母,然后交换a和b处的元素;当a没有遇到元音字母则继续向后遍历;b同理。
具体地,通过两个flag_a和flag_b来标记当前是否a和b遇到了元音字母:
若flag_a=0说明a尚未遇到,a++;flag_a=1则a不变
若flag_b=0说明b尚未遇到,b–; flag_b=1则b不变
若flag_a=1且flag_b=1说明a和b此时都为元音字母,交换,并移动a和b,进行a++,b–
直到下标超出范围,a>=b或b<0或a>=n,停止判断,返回字符串s。