PTA：编程实现strncpy函数功能（C语言）

题目

本题要求编程实现“n"族字符串拷贝函数功能strncpy()，而不使用标准库自带函数。

函数接口定义： char *MyStrNcpy(char *dst, const char *src, int n);

此函数把字符串src中的前n个字符复制到到dst所指向的内存空间。
（1）如果src指向的字符串少于n个字符，则将’\0’加到dst的尾部，直到满足n个字符为止；
（2）如果src指向的字符串大于n个字符，则拷贝src的前 n个字符到dst；
（3）返回指向dst的指针。

裁判测试程序样例：
在这里给出函数被调用进行测试的例子。例如：

#include 
#define N 10
char *MyStrNcpy(char *dst, const char *src, int n);
int main()
{
    char arr_dst[N] = "AAAAAAAAA";
    char arr_src[N] = { ‘\0’ };
        
    gets(arr_src);//输入源字符串
    
    int n;
    scanf("%d",&n);//输入拷贝长度
    
    char * new_dst;//函数调用返回目的字符串指针
    new_dst = MyStrNcpy(arr_dst, arr_src, n);
    
    printf("%s\n", new_dst);//打印输出
    return 0;
}

/* 请在这里填写答案 */

样例

输入样例1：
China
2
输出样例1：
ChAAAAAAA
输入样例2：
China
6
输出样例2：
China

解题

char *MyStrNcpy(char *dst, const char *src, int n) {  
    int i = 0;  
    while (i < n && src[i] != '\0') {  
        dst[i] = src[i];  
        i++;  
    }  
    while (i < n) {  
        dst[i] = '\0';  
        i++;  
    }  
    return dst;  
}