每日两道leetcode

345. 反转字符串中的元音字母 - 力扣（LeetCode）

题目

给你一个字符串 s ，仅反转字符串中的所有元音字母，并返回结果字符串。

元音字母包括 'a'、'e'、'i'、'o'、'u'，且可能以大小写两种形式出现不止一次。

示例 1：

输入：s = "IceCreAm"

输出："AceCreIm"

解释：

s 中的元音是 ['I', 'e', 'e', 'A']。反转这些元音，s 变为 "AceCreIm".

示例 2：

输入：s = "leetcode"

输出："leotcede"

提示：

• 1 <= s.length <= 3 * 105
• s 由 可打印的 ASCII 字符组成

思路

1. 两种思路，一种是将找到的元音字母入栈，再重新遍历遇到元音字母就弹栈修改；另一种是使用swap函数，利用首尾指针，将两个元音直接交换。（速度快一点，少一遍遍历）

代码实现

class Solution {
public:
    string reverseVowels(string s) {
        int i = 0, n = s.length()-1;
        unordered_map<char, int> vocal= {{'a',1}, {'e',1}, {'i',1}, {'o',1}, {'u',1}, {'A',1}, {'E',1}, {'I',1}, {'O',1}, {'U',1}};
        while(i <= n) {
            if(vocal.count(s[i]) && vocal.count(s[n])) {
                swap(s[i], s[n]);
                i++;
                n--;
            }
            else if(vocal.count(s[i])) n--;
            else if(vocal.count(s[n])) i++;
            else {
                i++;
                n--;
            }
        }
        return s;
    }
};

复杂度分析

• 时间复杂度：哈希查表的时间复杂度是O(1)的，经搜索swap在string和vector上的时间复杂度也是近似O(1)的，所以总的时间复杂度就是遍历复杂度，最坏时间复杂度近似为O(n)。
• 空间复杂度：存了元音字符的哈希集合O(1)。

函数学习

• unordered_map的列表初始化方法unordered_map umap = {{key1,value1}, {key2, value2}};
• swap(a, b)——针对元素的交换时间是O(1)的，针对数组的交换时间是O(n)的，数组交换的过程中会导致迭代器失效，需注意。