Question

我有一个用于数据挖掘类的项目，我必须为文件编写库的采样算法。该程序将数字k，输入文件的名称以及要创建的输出文件的名称作为输入。输出文件必须包含输入中的k个随机行。我尝试了一些东西，但输出错了。

这是我使用的代码：

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>

int countLines(FILE* file)
{
    char ch,lines=0;

    while ((ch=fgetc(file)) != EOF)
        if (ch=='\n')
            lines++;
    return(lines);
}

void itemSelection(FILE* fp1, FILE* fp2, int k)
{
    int i,j,n,test=0;
    char line[256];
    char** buffer;

    srand((unsigned int) time(NULL));
    buffer = (char**)malloc(sizeof(char*));
    for(i=0;i<k;i++)
        buffer[i] = (char*)malloc(256*sizeof(char));
    n = countLines(fp1);
    if(k>n)
    {
        rewind(fp1);
        while(fgets(line, 256, fp1)!=NULL)
        {
            printf("%s test\n",line); 
            fprintf(fp2,"%s",line); 
        }
    }
    else
    {
        rewind(fp1);    
        for(i=0;i<k;i++)
        {
            fgets(line, 256, fp1);
            buffer[i]=line;
            printf("first k lines:\t%s\n",buffer[i]);
        }
        for(i=k;i<n;i++)
        {
            fgets(line,256,fp1);
            printf("line is:\t%s.\n", line);
            j = rand() % (i+1);
            if(j<k)
            {
                buffer[j]=line;
                printf("later parts are:\t%s. J is:%d.\n",buffer[j],j);
            }
        }
    }
    for(i=0;i<k;i++)
        printf("buffer test:\t%s.\n", buffer[i]);
}

void printFunc(FILE* fp2,int k)
{
    char line[256];
    int i;

    rewind(fp2);
    for(i=0;i<k;i++)
    {
        fgets(line, 256, fp2);
        printf("print test is:\t%s.\n",line);
    }   
}

void main(int args, char** argv)
{
    FILE* fp1;
    FILE* fp2;
    int k;

    if(args<4)
    {
        printf("Expected more arguments!\n");
        exit(-1);
    }

    fp1 = fopen(argv[2],"r");
    if(fp1 == NULL)
    {
        printf("Could not open input file!\n");
        perror("Error: ");
        exit(-1);
    }
    fp2 = fopen(argv[3],"w");
    if(fp2 == NULL)
    {
        printf("Could not open output file!\n");
        perror("Error: ");
        exit(-1);
    }
    k = atoi(argv[1]);
    itemSelection(fp1, fp2, k);
    printFunc(fp2,k);
    fclose(fp1);
    fclose(fp2);
}

该程序尝试做的是从文件中读取k个第一行并将其存储在（k，256）大小的二维字符串数组中。然后，对于每个下一行生成一个随机数j，如果该数字小于k，则将缓冲区[j]替换为从文件中取出的最近一行。

然而，我得到的输出由k行}组成，这是输入的最后一个字符。像这样（例如k = 5）：

}
}
}
}
}

当我打印缓冲区以查看其内容时，它会正确显示。但是当我写入文件时，它写错了输出。

任何帮助都将受到高度赞赏！提前谢谢！

Answer 1

选择行时，必须复制行的内容，而不是指针（始终为line，其内容将不断被覆盖）：

buffer[i]=line;

应该是

strcpy(buffer[i], line);

同样适用于buffer[j]。

您还有分配错误：

buffer = (char**)malloc(sizeof(char*));

应该是：

buffer = malloc(k * sizeof(char*));

为k字符串留出空间（并注意有关casting the result of malloc in C的热门建议）。您还应该考虑使用buffer想要做什么：您是否将其返回，以便客户端代码可以使用它（并且必须free）或者它是itemSelection的本地代码，返回之前应该free。目前你什么都没做。

最后，在您的countLines函数中，变量应为int，而不是char：fgetc返回一个int，因此您可以在所有有效（无符号）之间进行排序char值和特殊值EOF。

水库采样算法不起作用

1 个答案: