O(n)时间原地删除指定字符串中的指定字符

声明： 原创作品，转载请注明出处!

问题：实现一个函数，要求将一个字符串里面的指定字符全部删除，最后返回经过修改后的字符串，要求不能调用任何其他C语言运行时函数，并且复杂度尽可能的低。

给定函数如下：(ch是指定被删除的字符)

char *delete_all(const char charset, char *str)
{
    /* write code here */
}

基本思路：

首先考虑用最简单的方式，设置两个指针s和q，s指向str的第一个位置，q指向一块新的空间，用以保存经删除指定字符后的字符串。那么，代码应该是这样的：

char *delete_all(const char ch, char *str)
{
    char* s = str;
    char* q;
    char* new_str;

    if(str == NULL){
        return str;
    }
 
    new_str = (char*) malloc(strlen(str)+1);
    q = new_str;
    while(*q = *s++){
        if(q != ch){
            ++q; 
        }
    }
    strcpy(str, new_str);
    free(new_str);
    return str;
}

虽然时间复杂度为O(n)，但是，上述代码的实现使用了C运行时函数，不符合题意。

仔细分析可以发现，s向前前进的步数 >= q向前移动的步数，如果将s和q指向同一块区域（因为修改后的字符串影响不到s指向的剩余未被处理的字符串），每次一步往前扫描并将遇到的字符ch赋给q当前指向的地址，如果ch为需要删除的字符，那么q就不动，如果不是要删除的字符，q就前进一步。那么，当扫描一遍后，得到的字符串就是删除了指定字符后的字符串了。

因此，有如下的代码：

char* delete_all(const char ch, char* str)
{
    if(str == NULL){
        return str;
    }

    char* s = str;
    char* q = str;
    while(*q = *s++){
        if(q != ch){
            ++q;
        }
    }

    return str;
}

很显然时间复杂度为 O(n)，空间复杂度为O(1)。