用較小的代價去除字符串中的空格

題目：

　　用C語言寫一個函數，去除字符串中的空格，並返回刪除的空格的個數。不允許開辟新的空間，只能申請簡單類型的自動變量。時間復雜度要求為O(n)。

比如：char str[]="dhkak   df d fd     fdjfkda     dfd   ff f  fd da ";

 

處理之后： str[]="dhkakdfdfdfdjfkdadfdffffdda";

返回刪除空格的個數為：12
算法思想：
　　先取字符串的長度，然后用前后各一個指針，分別為p,q,使前面一個指向空格，后面一個指向字符，然后用后面的值替換前面的空格，然后用空格替換；p指向下一個空格，q指向下一個值，直到p指向'\0'跳出循環，計算此時字符串的長度，再用處理之前的長度減去處理之后的長度，就可以得到刪除的空格的個數。
代碼如下：

#include <stdio.h>
int deleteSpace(char *pstr);
int main()
{
    char str[]="dhkak df d fd   fdjfkda  dfd ff  f fd da ";
    int spaces;
    spaces = deleteSpace(str);
    printf("we have removed %d spaces",spaces);
    return 0;
}
int deleteSpace(char *pstr)
{
    int len1 = strlen(pstr);
    char *p,*q;
    p=pstr;
    while(*p!=' ') p++;
    q=p;
    while(1)
    {
        while(*q==' '||*q!='\0') q++;
        if(*q=='\0')
        {
            *p=*q;
            break;
        }
        else
        {
            *p=*q;
            *q=' ';
            q++;
            p++;
        }
    }
    int len2 = strlen(pstr);
    int count = len1-len2;
    return count;
}

這樣就可以以時間復雜度為O(n)，實現這個函數。