程序员面试金典 1.4 空格替换

题目

请编写一个方法，将字符串中的空格全部替换为“%20”。假定该字符串有足够的空间存放新增的字符，并且知道字符串的真实长度(小于等于1000)，同时保证字符串由大小写的英文字母组成。
给定一个string iniString 为原始的串，以及串的长度 int len, 返回替换后的string。

我的题解

string replaceSpace(string iniString, int length) {
        // write code here
        int i = 0, count = 0, p;
        for(; i < length; ++ i)
            if(iniString[i] == ' ')
                count ++;

        p = 2 * count + length;

        char c_arr[3000];
        c_arr[p --] = '\0';

        for(i = length - 1; i >= 0; -- i) {
            if(iniString[i] != ' ') {
                c_arr[p --] = iniString[i];
            }else {
                c_arr[p --] = '0';
                c_arr[p --] = '2';
                c_arr[p --] = '%';
            }
        }

        return string(c_arr);
    }

这个题解我写的并不满意，因为令我满意的题解并不需要额外开辟数组，而是直接在原数组上进行修改，并不需要额外的空间。如下：

string replaceSpace(string iniString, int length) {
        // write code here
    int i = 0, count = 0, p;
    for(; i < length; ++ i)
        if(iniString[i] == ' ')
            count ++;

    p = 2 * count + length;

    if(iniString.size() < p)
        return "ERROR";

    iniString[p --] = '\0';

    for(i = length - 1; i >= 0; -- i) {
        if(iniString[i] != ' ') {
            iniString[p --] = iniString[i];
        }else {
            iniString[p --] = '0';
            iniString[p --] = '2';
            iniString[p --] = '%';
        }
    }

    return iniString;
}

题目中也说了，数组足够长，这样解才更好。但是牛客网OJ的测试用例似乎并不符合提议，所以想要通过只能重新开一个数组了。