PAT 字符串-02 删除字符串中的子串

/*
 2 *PAT 字符串-02 删除字符串中的子串
 3 *2015-08-09
 4 作者:flx413
 5 */
 
#include<stdio.h>
#include<string.h>

void fun(char *str, char *substr) {
    int i, j, k, flag;
    int len = strlen(substr);
    
    while(strstr(str, substr)) {                //如果字符串中存在子串
        flag = 0;                               //初始化 flag = 0
        
        for(i = 0; str[i]; i++) {   
            for(j = i, k = 0; substr[k] == str[j] && substr[k] != '\0'; k++, j++) {
                if(substr[k + 1] == '\0') {     //找到子串第一次出现的位置 i，flag 赋值为1
                   flag = 1;
                }
            }
            if(flag) {                          //对原来的字符串做删除
                for(; str[i + len]; i++) {
                    str[i] = str[i + len];
                }
                str[i] = '\0';
                break;
            }
        }
    }
}

int main() {
    char str[81],substr[81];                //题目要求输入字符串长度不大于80，定义str[80]会出错
    gets(str);
    gets(substr);
    fun(str, substr);
    printf("%s", str);
    return 0;
}