C primer plus 第十一章 练习1：

/*
===========================================================================
    设计并测试一个函数，可以从输入读取n个字符（包括字符、制表符和换行符），
把结果存储在一个数组中，这个数组的地址通过参数来传递。
===========================================================================
*/

#include <stdio.h>
#include <string.h>
char *str(int n, char *s);
int main(void)
{
    char s[5];
    char *pr = str(5, s);
    for (int i = 0;i < 5;i++) {
        putchar(pr[i]);
    }
    printf("\n");
    return 0;
}

char *str(int n, char *s)
{
    int i = 0;
    char a;
    printf("请输入%d个字符： ", n);
    while ((a = getchar()) != '\0' && i < n) {
        s[i] = a;
        i++;
    }
    return s;
}