C/C++函数返回值为指针时（2016美团模拟笔试）

来源：互联网发布：乐乎公寓公益西桥编辑：程序博客网时间：2024/05/20 10:52

昨天美团模拟笔试，最后一个编程题：输入一个字符串，删去多余空格

如输入：" I love China ! "
输出："I love China !"
题目给了一个函数框架：
char* getString(char * sourceString)
{//待填充}

按理说这题并不难，个人感觉重点就在于返回值怎么确定。

1）如果队员指针进行处理，直接返回当然是没有问题的咯。

只是在删除空格之后的操作相对复杂一点。

2）在函数内部声明一个新指针，然后返回。

那么问题就出来了：

a 最简单最不过大脑的办法就是
char a[10]; 完成操作后最后 return a;

获取可以得到正确结果，那是因为指针指向的内存暂时没有其他用处。

函数体中声明的为局部变量，在函数运行后会被自动撤回，这种方法是错误的。

b 其实我们可以在函数体内部通过动态申请（malloc/new）
char *a = (char*)malloc(size_t); 操作之后在return a;

切记这里a最终是要释放的，我们可以在调用该函数的函数体中free/delete。

鄙人写了一段代码，欢迎吐槽及纠错：

#include <iostream>using namespace std;char *getString(char * sourceString){if(NULL == sourceString){return NULL;}int length = strlen(sourceString);//length = sizeof(sourceString); 这里得到的长度是4，是指针的长度char *a = (char *)malloc(sizeof(char)*length);int flag = 1;     //避免第一个出现空格的情况int count = 0;while(length > 0){if(' ' == *sourceString){if(0 == flag){*(a + count) = *sourceString;count++;}flag = 1;}else{*(a + count) = *sourceString;count++;flag = 0;}sourceString++;length--;}//处理最后一个空格，并在结尾加上结束符if(' ' == *(a + count - 1)) {*(a + count - 1) = '\0';}else{*(a + count) = '\0';}return a;}int main(){char a[20] = " i love china ";//int len = strlen(a);    这里得到的长度是14。//len = sizeof(a);        这里得到的长度是20。char *b;b = getString(a);cout << strlen(b) << endl;cout << b << endl;if(NULL != b){free(b);}system("pause");return 0;}

0 0