Подсчёт количества символов и замена

#include <stdio.h>
#include <ctype.h>
#include <string.h>
#include <stdlib.h>//malloc
 
int poisk_AB(char *str);
char *zamena(char *str);
 
int main()
{
    char text[1024] = {0};//Áóôôåð äëÿ ââîäèìîãî òåêñòà
    printf("Enter text : ");
    scanf("%[^\n]", text);//Ââîä ñòðîêè ñ ïðîáåëàìè
    printf("Num chars among A-B : %d\n", poisk_AB(text));
    char * ptr = zamena(text);
    printf("String after replacing 0 : \n");
    if(ptr == NULL)
        printf("No replacing had been take\n");
    else
    {
        printf("%s\n", ptr);
        free((void *)ptr);
    }
    return 0;
}
 
int poisk_AB(char *str)
{
    int i      = 0;
    int FlagA  = 0;//Áóäåò îçíà÷àòü ÷òî â ñòðîêå íàéäåí ñèìâîë À
    int FlagB  = 0;//Áóäåò îçíà÷àòü ÷òî â ñòðîêå íàéäåí ñèìâîë Â
    int nCount = 0;
    if(str == NULL)
        printf("Empty string as param poisk_AB!\n");
    else
    {
        for(i = 0; i < strlen(str); i = i + 1)
        {
            if(toupper(str[i]) == 'A')
                FlagA = 1;
            if(toupper(str[i]) == 'B')
                FlagB = 1;
            if( FlagA && FlagB == 0)
                nCount = nCount + 1;
        }
        if(FlagA == 0)
            printf("String not contain symbol A!\n");
        else
        if(FlagB == 0)
            printf("String not contain symbol B!\n");
    }
    return nCount;
}
 
char *zamena(char *str)
{
    int i      = 0;
    int j      = 0;
    char * buf = 0;
    int nCount = 0;
    if(str == NULL)
        printf("Empty string as param poisk_AB!\n");
    else
    {
        //Ñ÷èòàåì ñêîëüêî çàìåí íàäî ïðîèçâåñòè
        for(i = 0; i < strlen(str); i = i + 1)
        {
            if(str[i] == '0')
                nCount = nCount + 1;
        }
        //Âûäåëÿì áîëüøóþ ïàìÿòü (áîëüøóþ íà ÷èñëî çàìåí)
        buf = (char *)malloc((strlen(str) + nCount + 1)*sizeof(char));
        if(buf == NULL)
            printf("Allocation memory error\n");
        else
        {
            for(i = (j = 0); i < strlen(str); i = i + 1)
            {
                if(str[i] == '0')
                {
                    buf[j]    = 'O';
                    buf[j + 1]= 'O';
                    j = j + 1;
                }
                else
                    buf[j] = str[i];
                j = j + 1;
            }
            buf[j] = 0;
        }
    }
    return buf;
}