算法17：实现memcpy(void *src, int size, void *dest)

算法分析

这道题主要考察一下几点
1、对于参数非法的判断，空指针的判断和size大小的判断
2、源数据不能被改写
3、源数据和目标数据是否存在内存重叠

c++代码

不考虑内存重叠的情况

char* memcpy(char* dst,const char* src,size_t count)
{
    char* ptr = dst;
    if(dst == NULL || src == NULL || count <= 0)
    {
        return NULL;
    }

    while(count--)
    {
        *dst++ = *src++;
    }

    //不能返回dst的，因为dst指针被++了
    return ptr;
}

考虑内存重叠的情况

char* memcpy(char* dst,const char* src,size_t count)
{
    char* ptr = dst;
    if(dst == NULL || src == NULL || count <= 0)
    {
        return NULL;
    }

    //判断是否存在内存重叠
    if(dst <= src || dst >= src+count)
    {
        //低地址先拷贝
        while(count--)
        {
            *dst++ = *src++;
        }
    }
    else
    {
        //高地址先拷贝
        dst = dst + count -1;
        src = src + count -1;
        while(count--)
        {
            *dst-- = *src--;
        }
    }

    return ptr;
}

注意事项

函数参数count为size_t,是无符号类型，判断无符号数<0是没有任何意义的，如果传进来-1的数，会产生一个很大的无符号整型数据。所以要么在调用函数之前判断count是否为负数，要么memcpy参数由size_t改成int类型。