6-1 字符串 - 8. 输入字符串 (10 分)

C语言标准函数库中 scanf 函数和 gets 函数都可以输入字符串，但是各有优、缺点。我们综合两者的优点，克服两者的缺点，设计一个函数来输入字符串。

函数原型

char* GetStr(char *str, int size);

说明：str 为字符串的起始地址，size 为字符数组的尺寸。函数读取用户从键盘输入的字符串(以换行符 '\n' 结束)到 str 所指示的字符数组中，并在字符末尾添加字符串结束标记 '\0'，函数值为 str。显然，字符串的最大长度为 size - 1，为字符串结束标记 '\0' 预留空间。若用户输入的字符过多，则函数最多读取 size - 1 个字符，剩余字符仍留在缓冲区中，可以继续被后面的输入函数读取。

裁判程序

#include <stdio.h>

char* GetStr(char *str, int size);

int main()
{
    char a[10], b[10];
    GetStr(a, 10);
    GetStr(b, 10);
    puts(a);
    puts(b);
    return 0;
}

/* 你提交的代码将被嵌在这里 */

输入样例1

Bob
Mary

输出样例1

Bob
Mary

输入样例2

Constantine

输出样例3

Constanti
ne

输入样例3

Francisco
Stevenson

输出样例3

Francisco
Stevenson

提示：可利用 ungetc 将最后一个字符退回缓冲区。

代码如下：

char* GetStr(char *str, int size)
{
    int i = 0;
    char c;
    while(c = getchar(),c != '\n'&& i < size - 1)
    {
        str[i] = c;
        ++i;
    }
    str[i] = '\0';
    if(c != '\n') ungetc(c,stdin);
    return str;
}