[NewCode 4] 替换空格

题目描述

请实现一个函数，将一个字符串中的空格替换成“%20”。例如，当字符串为We Are Happy.则经过替换之后的字符串为We%20Are%20Happy。

最直接的方式，直接使用 STL。假设字符串的长度是n，那么对于每个空格字符，需要移动后面 o(n) 个字符，因此对含有 o(n) 个空格字符的字符串而言时间复杂度为 o(n²)，过高了。代码如下：

class Solution {
public:
    string replaceSpace(string str) {
        size_t ix = 0; 
        while (ix < str.size()) {
            if (str[ix] == ' ') {
                str.replace(ix, 1, "%20");
                ix += 3;
            } else {
                ix++;
            }  
        }
        return str;
    }
};

比较好的方式，是从后往前移动，如此的时间复杂度是 o(n) 。来看代码：

class Solution {
public:
	string replaceSpace(string str) {
        if (str.empty()) {
            return str;
        }
        
        int preSize = str.size();
        // 统计空格数量
        int spaceCnt = 0;
        for (const auto &character : str) {
            if (character == ' ') {
                spaceCnt++;
            }
        }
        
        // 为str重新分配空间
        int curSize = preSize + (3 - 1) * spaceCnt;
        str.resize(curSize);
        
        // 替换
        int prePos = preSize - 1;
        int curPos = curSize - 1;
        while (prePos >= 0) {
            if (str[prePos] == ' ') {
                str[curPos--] = '0';
                str[curPos--] = '2';
                str[curPos--] = '%';
            } else {
                str[curPos--] = str[prePos];
            }
            
            prePos--;
        }
        
        return str;

	}
};