Exe7 刪除字串中指定的字元 （還有有問題——指標的問題）

#include <iostream>
using namespace std;

 

//在一個含有不重複的字串中尋找一個字元的位置，不存在返回NULL
char* findc(char* p, char c)
{
 int flag = 0; //預設不存在
 while(*p++)
 {
  if(*p==c)
  {
   flag = 1;
   break;
  }
 }
 if(flag)
  return p;
 else
  return NULL;
}

char* DeleteStr(char* str,char* substr)
{
 char* head = NULL;
 char* p = NULL;
 if(str==NULL)
  return NULL;
 head = p = str;
 while(*p++)
 {   //*p不在substr內部，則必然在str中
  if(findc(substr,*p)==NULL)
   *str++ =*p;
 }
 *str = '/0'; //最後加上字串結束符
 return head;
}

int main()
{
 char str[128];
 char substr[20];
 gets(str);
 gets(substr);
 char* result = DeleteStr(str,substr);
 cout<<result<<endl;
 return 0;
}