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)。