力扣比特位计数C语言

Question

快要疯掉了，到底哪里有错误啊，为什么memset(dp,0,sizeof(dp))完了之后dp[2..]之后都不是0，最后结果错误

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* countBits(int n, int* returnSize) {
    int *dp=(int *)malloc(sizeof(int)*100001); 
    *returnSize=n+1;//规定返回数组的长度，你只返回了指针，所以需要这个来让外部知道你的数组有多长
    //该题需要return一个数组，而在不申请空间（不malloc）的情况下函数内建立的数组是局部变量，无法带回主函数
    memset(dp,0,sizeof(dp));
    for(int i=0;i<=n;i++)
        printf("%d ",dp[i]);
       
    int i;
    for(i=1;i<=n;i++)
    {
        int demp=i;
        while(demp)
        {
            int n=demp%2;
            demp=demp/2;
            if(n==1) 
                dp[i]++;
        }
    }
    return dp;
}

Answer 1

该回答引用自GPT-3.5,由博主GISer Liu编写：

根据您提供的代码，问题可能出在 memset(dp, 0, sizeof(dp)) 上。memset
是按字节进行内存设置的，而 dp 是一个指针，它所指向的是整数数组，每个整数占用 4 个字节（32 位系统），所以在使用 memset 初始化时，应该传入的是要初始化的字节数，而不是数组元素个数。因此，正确的用法应该是 memset(dp, 0, sizeof(int) * 100001)。
此外，在您的代码中，使用了两个同名的变量 n，这可能导致逻辑错误，建议将其中一个变量名改为其他名称，以避免混淆。
以下是修改后的代码：
cCopy code

int* countBits(int n, int* returnSize) {
int *dp = (int *)malloc(sizeof(int) * 100001);
*returnSize = n + 1;
memset(dp, 0, sizeof(int) * 100001);
for(int i = 1; i <= n; i++) {
int demp = i;
while(demp) {
int bit = demp % 2;
demp = demp / 2;
if(bit == 1)
dp[i]++;
}
}
return dp;
}

请注意，在使用完 dp 数组后，记得在适当的时候调用 free(dp) 释放内存，以免造成内存泄漏。

如果该回答解决了您的问题，请采纳！如果没有，请私信联系或评论您的疑惑