声明: 原创作品,转载请注明出处!
问题:实现一个函数,要求将一个字符串里面的指定字符全部删除,最后返回经过修改后的字符串,要求不能调用任何其他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)。