算法学习(七)有内存限制的海量数据排序

磁盘文件排序

问题描述:
输入:给定一个文件,里面最多含有n个不重复的正整数(也就是说可能含有少于n个不重复正整数),且其中每个数都小于等于n,n = 10^7。
输出:得到按从小到大升序排列的包含所有输入的整数的列表。
条件:最多有大约1MB的内存空间可用,但磁盘空间足够。且要求运行时间在五分钟以下,10秒为最佳结果。
分析:
首先注意的是它的内存要求,基本上否决了大多数的排序算法。
文件里面的数据是不重复的,这个很关键,因为下面的方案前提就是数据不重复。

位图方案

以位为最小单位,对应位为1,代表位数值,可以用20位长的字符串来表示一个所有元素都小于20的简单的非负数集合,边框用如下字符串来表示集合{1,2,3,5,8,13}:
01110100100001000000
对应位置则置1,没用对应数的位置则置0,开始扳手指头数数吧。我们可以采用一个具有1000万个位的字符串来表示这个文件,其中,当整数i在文件中时,第i位为1。
第一步,所有位置0,将集合初始化。
第二步,通过读入文件中的每个整数来建立集合,将对应的位都置1.
第三步,检验每一位,如果该位为1,就输出对应的整数。
这里不得不提一下c++库中的一个容器bitset,发现这个在处理位的时候简直牛,先来瞅瞅。

#include <iostream>
#include <bitset>
using namespace std;
int main()
{
    const int max = 20;
    int arr[10] = {2,9,3,5,4,0,18,11,12,15};
    bitset<max> bit_map;  //一种类模板,放入长度,定义了一个max位,每位都为0
    bit_map.reset();  //所有位置0
    int i;
    for(i = 0;i< 10;i++)
    {
        bit_map.set(arr[i]);  //将对应位置1
    }
    for(i = 0;i< max;i++)
    {
        if(bit_map.test(i))  //判断第i位,如果为1,返回true
            cout << i << " ";
    }
    cout << endl;
    return 0;
}

再加上文件操作,就是针对海量数据排序。

#include <iostream>
#include <bitset>
#include <assert.h>
#include <time.h>
#include <stdio.h>
#include <stdlib.h>
using namespace std;

const int MAX = 5000000;
int main()
{
    clock_t begin = clock();
    bitset<MAX> bit_map;
    bit_map.reset();

    FILE * fp_unsort_file = fopen("data.txt","r");
    assert(fp_unsort_file);
    int num;
    while(fscanf(fp_unsort_file,"%d ",&num)!= EOF)
    {
        if(num < MAX)   //先对1-4999999的数进行排序
            bit_map.set(num);
    }
    FILE * fp_sort_file = fopen("sort.txt","w");
    assert(fp_sort_file);
    int i;
    for(i= 0;i < MAX;i++)
    {
        if(bit_map.test(i))
            fprintf(fp_sort_file,"%d ",i);
    }
    int result = fseek(fp_sort_file,0,SEEK_SET); //指针回到文件开始处
    if(result)
        cout << "fseek failed" << endl;
    else
    {
        bit_map.reset();
        while(fscanf(fp_unsort_file,"%d ",&num) !=EOF)
        {
            if(num > MAX && num < 10000000)  //再对5000000-10000000的数排序
            {
                num -=MAX;
                bit_map.set(num);
            }
        }
        for(i = 0;i<MAX;i++)
        {
            if(bit_map.test(i))
                fprintf(fp_sort_file,"%d ",i+MAX);
        }
    }
    clock_t end = clock();
    cout << "time is:" << endl;
    cout << (end-begin)/CLOCKS_PER_SEC << "s" << endl;
    fclose(fp_sort_file);
    fclose(fp_unsort_file);
    return 0;
}

第一次只处理1-4999999之间的数据,对这些数进行位图排序,只需要约5000000/8 = 625000byte,就是0.625M,排序后输出。
第二次,扫描输入文件,只处理4999999-10000000的数据项,只需要0.625M,因此只需要0.625M。

多路归并

我们先了解一下,归并排序的过程,归并排序就是2路归并
归并排序的过程:
1,把无序表的每一个元素看做是一个有序表,则有n个有序子表;
2,把n个有序表按相邻位置分成若干对,每对中的两个表进行归并,归并后子表数减少一半。
3,反复进行这一过程,直到归并为一个有序表为止。
是采用分治法的典型应用
首先考虑下如何将二个有序数列合并,比较二个数列第一个数,谁小就放入到临时数组中,然后再进行比较,如果一个数列到达结尾,直接将另一个数列放入就临时数组中。

oid MemeryArray(int a[],int n,int b[],int m,int c[])
{
    int i,j,k;
    i = j= k = 0;
    while(i < n && j< m)
    {
        if(a[i] < b[j])
            c[k++] = a[i++];
        else
            c[k++] = b[j++];
    }
    while(i < n)
        c[k++] = a[i++];
    while(j < n)
        c[k++] = b[j++];
}

下一步考虑如何让两个A,B数组有序,可以将A,B各自再分成二组,依次类推,当分出来的小组只有一个元素时,这个小组内已经达到有序了,然后再合并相邻的二个小组就可以了,先递归分解,再合并数列就完成了归并排序。

//将二个有序数列a[first...mid] 和a[mid+1...last]合并
void MemeryArray(int a[],int first,int mid,int last,int temp[])
{
    int i  = first;
    int j = mid+1;
    int m = mid;
    int n = last;
    int k = 0;
    while(i <= m && j< =n)
    {
        if(a[i] < b[j])
            temp[k++] = a[i++];
        else
            temp[k++] = a[j++];
    }
    while(i <= m)
        temp[k++] = a[i++];
    while(j <=n)
        temp[k++] = a[j++];
    for(i = 0;i<k;i++)
        a[first+i] = temp[i];
}
void mergesort(int a[],int first,int last,int temp[])
{
    if(first < last)
    {
        int mid = (first + last)/2;
        mergesort(a,first,mid,tmep);
        mergesort(a,mid+1,last,temp);
        MemeryArray(a,first,mid ,last,temp);
    }
}

归并排序是一种稳定排序,时间复杂度最好情况下和最坏情况下均是O(nlogn)
多路归并就是从多个有序数列中归并。
比如将10000000的数据,分成40个有序文件,分别在内存中排序,然后对这40个有序文件进行归并排序。
1,读取每个文件中第一个数(每个文件的最小数),存放在一个大小为40的data数组中,
2,选择data数组中最小的数min_data,及其相应的文件索引(来自哪个文件)index
3,将min_data写入到文件result,然后更新数组data(根据index,读取该文件的下一个数代替min_data)
4,判读是否所有数据都读取完毕,否则返回到2步。

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

using namespace std;

int sort_num = 10000000;
int memory_size = 250000;  //每次排序250000个数据

int read_data(FILE *fp,int *space)
{
    int index = 0;
    while(index < memory && fscanf(fp,"%d ",&space[index])!= EOF)
        index++;
    return index;
}
void write_data(FILE * fp,int *space,int num)
{
    int index = 0;
    while(index < num)
    {
        fprintf(fp,"%d ",space[index]);
        index++;
    }
}
void check_fp(FILE *fp)
{
    if(fp == NULL)
    {
        cout << "the file not open" << endl;
        exit(1);
    }
}
int compare(const void * a,const void * b)
{
    return *(int *)a-*(int *)b;
}
string new_filename(int n)
{
    char file_name[20];
    sprintf(file_name,"data%d.txt",n);
    return file_name;
}
//将原文件分割成多个小文件,并在内存中排序
int memroy_sort()
{
    FILE * fp_in_file = fopen("data.txt","r");
    check_fp(fp_in_file);
    int counter = 0;
    while(true)
    {
        int * space = new int(memory_size);
        int num = read_data(fp_in_file,space);
        if(num == 0)
            break;
        qsort(space,num,sizeof(int),compare);
        string file_name = new_filename(++counter);
        FILE *fp_aux_file = fopen(file_name,"w");
        check_fp(fp_aux_file);
        write_data(fp_aux_file,space,num);
        fclose(fp_aux_file);
        delete []space;
    }
    fclose(fp_in_file);
    return counter;
}
//将n个小文件归并为一个文件,多路归并
void merge_sort(int file_num)
{
    if(file_num < 0)
        return;
    FILE *fp_out_file = fopen("result.txt","w");
    check_fp(fp_out_file);
    FILE ** fp_array = new FILE*[file_num];  //存放每个小文件的指针的数组
    int i;
    for(i = 0;i<file_num;i++)
    {
        string file_name = new_file_name(i+1);
        fp_array[i] = fopen(file_name,"r");
        check_fp(fp_array[i]);
    }
    int * first_data  = new int [file_num];
    bool * finish  = new bool[file_num];  //文件是否读取到结尾
    memset(finish,false,sizeof(bool) * file_num);

    for(i = 0;i<file_num;i++)
        fscanf(fp_array[i],"%d ",&first_data[i]);

    while(true)
    {
        int index = 0;
        while(index < file_num && finish(index))
            index++;
        if(index >=file_name)
            break;
        int min_data = first_data[index];
        for(i = index+1;i<file_name,i++)
        {
            if(min_data > first_data[i] && !finish[index])
            {
                min_data = first_data[i];
                index = i;
            }
        }
        fprintf(fp_out_file ,"%d ",min_data);  //将最小的输出
        if(fscanf(fp_array[index],"%d ",&first_data[index]) == EOF) //根据index,找到该文件的下一个数,放入到first_data中。如果这个文件已经读取到结尾,则将标志置为true。
            finish(index) = true;
    }

    fclose(fp_out_file);
    delete []finish;
    delete[]first_data;
    for(i = 0;i<file_num;i++)
        fclose(fp_array[i]);
    delete[] fp_array;

}
int main()
{
    clock_t start = clock();
    int aux_file_num = memory_sort();
    clock_t memory_end = clock();
    cout << "time need in memory sort" << memory_end-start << endl;
    merge_sort(aux_file_num);
    clock_t end = clock();
    cout << "the time need in merge sort" << end - memory_end << endl;
    return 0;
}
qsort()用法

int cmp(const void* a,const void * b);
void qsort(char *,int n,int m,cmp);
包含在stdlib.h中,函数四个参数,
参与排序的数组,元素个数,单个元素到小(sizeof),比较函数。
再说比较函数,返回正数,a要放在b后面,返回负数,a要放在b前面,返回0,相等。

解决问题的关键是在于熟悉一个算法,而不是某一问题,
通过这个问题,我们学到了两种方法:
一种:位图法,
针对与不重复数据排序,进行数据的快速查找,判重,删除。
二中:多路归并
海量数据,内存有限的情况下。

    原文作者:排序算法
    原文地址: https://blog.csdn.net/shitangdejiaozi/article/details/51240107
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞