H面試程式3：去除字元串中多餘的空格

比如“_ _I_ _ _Lu_ _ _  _you_ _ _ _ _ _”變成“I_ Lu_you”

#include<stdio.h>
#include<assert.h>
//50min
void RemoveExtraSpace(char* str)
{
   assert(str);
   int flag = 1; //之前已經遇到空格
   int newp = 0;
   int i = 0;

   while(str[i] !='\0')
   {
  
	   if(flag &&str[ i] == ' ')//之前已經遇到空格,且現在又遇到空格
	   {
		   i++;
	   }
	   else if(flag &&str[i] != ' ')//之前已經遇到空格,現在沒遇到空格
	   {
		   str[newp++] = str[i++];
		   flag = 0;
		   
	   }
	   else if(flag == 0 &&str[ i] == ' ')//之前沒遇到空格,現在遇到空格
	   {
		   
		   str[newp++] = str[i++];
		   flag = 1;
	   }
       else //之前沒遇到空格,現在沒遇到空格
	   {
           str[newp++] = str[i++];
		   flag = 0;
	   }
   }
   if(str[newp-1] = ' ')   //處理結尾處多餘的空格
    str[newp - 1] = '\0';
   else
    str[newp] = '\0';
}


int main()
{
	char a[] = " I  LU  YOU        ";
    RemoveExtraSpace( a );
    return 0;
}