s_gets - s_gets_c bmstu

#include "pch.h"
#include <iostream>

char* s_gets(char* str, int size);
char* s_gets_c(char* str, int size);



int main()
{

    const int size = 50;
    char str[size];
    s_gets(str, size);
    s_gets_c(str, size);
    printf("%s\n", str);
    return 0;
}


char* s_gets(char* str, int size)
{
    if (fgets(str, size, stdin) == NULL)
    {
        return NULL;

    }
    int i = 0;
    int flag = 1;
    while (str[i] != '\0')  // hello world\n1234\0
    {
        if (str[i] == '\n')
        {
            flag = 0;
            break;
        }
    i++;

    }
    str[i] = '0';
    if (flag)
    {
        while (getchar() != '\n');
    }
    return str;
}


char* s_gets_c(char* str, int size)
{
    int  i = 0;
    char ch = '\0';
    while (( ch = getchar()) != '\n')
    {
        if (i < size)
        {

            str[i] = ch;
        }
            i++;

    }
    str[i] = '\0';
        return str;
}