分享

两个含有大量数据的文件中寻找相同数据的各种算法对比

1.面对两个数据量文件比较,你所想到的方法是什么?
你的方法是否具有下面特点
1.考虑内存泄漏
2.代码规范,考虑出错
3.具有可重用性
2.比较下面三种算法的区别?
3.什么是位图法?




求解问题如下:
在本地磁盘里面有file1和file2两个文件,每一个文件包含500万条随机整数(可以重复),最大不超过2147483648也就是一个int表示范围。要求写程序将两个文件中都含有的整数输出到一个新文件中。
要求:

程序的运行时间不超过5秒钟。
没有内存泄漏。
代码规范,能要考虑到出错情况。
代码具有高度可重用性及可扩展性,以后将要在该作业基础上更改需求。
初一看,觉得很简单,不就是求两个文件的并集嘛,于是很快写出了下面的代码。

  1. #include<iostream>
  2. #include<vector>
  3. #include<cstdlib>
  4. #include<algorithm>
  5. #include<fstream>
  6. using namespace std;
  7. void merge(const vector<int> &, const vector<int>&, vector<int> &);
  8. // Author: 397090770
  9. // Email: wyphao.2007@163.com
  10. // blog: http://www.iteblog.com
  11. // 仅用于学习交流,转载请务必留下这些注释。
  12. int main(){
  13.         vector<int> v1, v2;
  14.         vector<int> result;
  15.         char buf[512];
  16.         FILE *fp;
  17.         fp = fopen("file1", "r");
  18.         if(fp < 0){
  19.                 cout<<"Open file failed!\n";
  20.                 exit(1);
  21.         }
  22.         while(fgets(buf, 512, fp) != NULL){
  23.                 v1.push_back(atoi(buf));
  24.         }
  25.         sort(v1.begin(), v1.end());
  26.         fclose(fp);
  27.         fp = fopen("file2", "r");
  28.         if(fp < 0){
  29.                 cout<<"Open file2 failed!\n";
  30.                 exit(1);
  31.         }
  32.         while(fgets(buf, 512, fp) != NULL){
  33.                 v2.push_back(atoi(buf));
  34.         }
  35.         sort(v2.begin(), v2.end());
  36.         cout<<v1[v1.size() - 1]<<endl;
  37.         cout<<v2[v2.size() - 1]<<endl;
  38.         fclose(fp);
  39.         merge(v1, v2, result);
  40.         cout<<result.size();
  41.         ofstream output;
  42.         output.open("result");
  43.         if(output.fail()){
  44.                 cerr<<"crete file failed!\n";
  45.                 exit(1);
  46.         }
  47.         vector<int>::const_iterator p = result.begin();
  48.         for(; p != result.end(); p++){
  49.                 output<<*p<<endl;
  50.         }
  51.         output.close();
  52.         return 0;
  53. }
  54. void merge(const vector<int>& v1, const vector<int>& v2, vector<int> &result){
  55.         vector<int>::const_iterator p1, p2;
  56.         p1 = v1.begin();
  57.         p2 = v2.begin();
  58.         while((p1 != v1.end()) && p2 != v2.end()){
  59.                 if(*p1 < *p2){
  60.                         p1++;
  61.                 }else if(*p1 > *p2){
  62.                         p2++;
  63.                 }else{
  64.                         result.push_back(*p1);
  65.                         p1++;
  66.                         p2++;
  67.                 }
  68.         }
  69. }
复制代码
编译运行:
34631_o.gif
一看,不行,不满足上面的5秒之内,于是又想了很久,上面不是显示sys调用花了很长时间嘛,于是有写了一个程序,用快速排序+二分查找法实现,代码如下:
  1. #include <iostream>
  2. #include <fstream>
  3. #include <vector>
  4. #include <cstdlib>
  5. #include <cstdio>
  6. #define MAXLINE 32
  7. using namespace std;
  8. void qsort(vector<int>&, int, int);
  9. int partition(vector<int>&, int, int);
  10. bool binarySearch(const vector<int>&, int);
  11. // Author: 397090770
  12. // Email: wyphao.2007@163.com
  13. // blog: http://www.iteblog.com
  14. // 仅用于学习交流,转载请务必留下这些注释。
  15. int main(){
  16.         vector<int> v1, result;
  17.         int temp;
  18.         char buf[MAXLINE];
  19.         FILE *fd;
  20.         fd = fopen("file1", "r");
  21.         if(fd == NULL){
  22.                 cerr<<"Open file1 failed!\n";
  23.                 exit(1);
  24.         }
  25.         while(fgets(buf, MAXLINE, fd) != NULL){
  26.                 v1.push_back(atoi(buf));
  27.         }
  28.         fclose(fd);
  29.         //cout<<v1.size()<<endl;
  30.         qsort(v1, 0, v1.size() - 1);
  31.         /*vector<int>::const_iterator p = v1.begin();
  32.         for(; p != v1.end(); p++){
  33.                 cout<<*p<<endl;
  34.                 sleep(1);
  35.         }*/
  36.         fd = fopen("file2", "r");
  37.         if(fd == NULL){
  38.                 cerr<<"open file2 failed!\n";
  39.                 exit(1);
  40.         }
  41.         while(fgets(buf, MAXLINE, fd) != NULL){
  42.                 temp = atoi(buf);
  43.                 if(binarySearch(v1, temp)){
  44.                         result.push_back(temp);
  45.                 }
  46.         }
  47.         cout<<result.size();
  48.         return 0;
  49. }
  50. void qsort(vector<int> &v, int low, int hight){
  51.         if(low < hight){
  52.                 int mid = partition(v, low, hight);
  53.                 qsort(v, low, mid - 1);
  54.                 qsort(v, mid + 1, hight);
  55.         }
  56. }
  57. int  partition(vector<int> &v, int min, int max){
  58.         int temp = v[min];
  59.         while(min < max){
  60.                 while(min < max && v[max] >= temp)
  61.                         max--;
  62.                 v[min] = v[max];
  63.                 while(min < max && v[min] <= temp)
  64.                         min++;
  65.                 v[max] = v[min];
  66.         }
  67.         v[min] = temp;
  68.         return min;
  69. }
  70. bool binarySearch(const vector<int> &v, int key){
  71.         int low, hight, mid;
  72.         low = 0;
  73.         hight = v.size() - 1;
  74.         while(low <= hight){
  75.                 mid = (low + hight) /2;
  76.                 if(v[mid] == key){
  77.                         return true;
  78.                 }else if(v[mid] < key){
  79.                         low = mid + 1;
  80.                 }else{
  81.                         hight = mid - 1;
  82.                 }
  83.         }
  84.         return false;
  85. }
复制代码

正乐着呢,编译运行:
34634_1.gif
结果发现,user时间是2.194秒,整个时间还要比以前长,显然这种方法还是不行,原因就是两个文件太大了,500万条,不是一般小,且上面花的时间主要用在排序上面去了,于是就想,能不能不用排序完成?这时有个朋友和我说了一下位图法,灵感一来,自己又去改写了代码:
  1. #include <iostream>
  2. #include <cstdlib>
  3. #include <cstdio>
  4. #include <cstring>
  5. #include <fstream>
  6. #include <string>
  7. #include <vector>
  8. #include <algorithm>
  9. #include <iterator>
  10. #define SHIFT 5
  11. #define MAXLINE 32
  12. #define MASK 0x1F
  13. using namespace std;
  14. // Author: 397090770
  15. // Email: wyphao.2007@163.com
  16. // blog: http://www.iteblog.com
  17. // 仅用于学习交流,转载请务必留下这些注释。
  18. void setbit(int *bitmap, int i){
  19.         bitmap[i >> SHIFT] |= (1 << (i & MASK));
  20. }
  21. bool getbit(int *bitmap1, int i){
  22.         return bitmap1[i >> SHIFT] & (1 << (i & MASK));
  23. }
  24. size_t getFileSize(ifstream &in, size_t &size){
  25.         in.seekg(0, ios::end);
  26.         size = in.tellg();
  27.         in.seekg(0, ios::beg);
  28.         return size;
  29. }
  30. char * fillBuf(const char *filename){
  31.         size_t size = 0;
  32.         ifstream in(filename);
  33.         if(in.fail()){
  34.                 cerr<< "open " << filename << " failed!" << endl;
  35.                 exit(1);
  36.         }
  37.         getFileSize(in, size);
  38.         char *buf = (char *)malloc(sizeof(char) * size + 1);
  39.         if(buf == NULL){
  40.                 cerr << "malloc buf error!" << endl;
  41.                 exit(1);
  42.         }
  43.         in.read(buf, size);
  44.         in.close();
  45.         buf[size] = '\0';
  46.         return buf;
  47. }
  48. void setBitMask(const char *filename, int *bit){
  49.         char *buf, *temp;
  50.         temp = buf = fillBuf(filename);
  51.         char *p = new char[11];
  52.         int len = 0;
  53.         while(*temp){
  54.                 if(*temp == '\n'){
  55.                         p[len] = '\0';
  56.                         len = 0;
  57.                         //cout<<p<<endl;
  58.                         setbit(bit, atoi(p));
  59.                 }else{
  60.                         p[len++] = *temp;
  61.                 }
  62.                 temp++;
  63.         }
  64.         delete buf;
  65. }
  66. void compareBit(const char *filename, int *bit, vector<int> &result){
  67.         char *buf, *temp;
  68.         temp = buf = fillBuf(filename);
  69.         char *p = new char[11];
  70.         int len = 0;
  71.         while(*temp){
  72.                 if(*temp == '\n'){
  73.                         p[len] = '\0';
  74.                         len = 0;
  75.                         if(getbit(bit, atoi(p))){
  76.                                 result.push_back(atoi(p));
  77.                         }
  78.                 }else{
  79.                         p[len++] = *temp;
  80.                 }
  81.                 temp++;
  82.         }
  83.         delete buf;
  84. }
  85. int main(){
  86.         vector<int> result;
  87.         unsigned int MAX = (unsigned int)(1 << 31);
  88.         unsigned int size = MAX >> 5;
  89.         int *bit1;
  90.         bit1 = (int *)malloc(sizeof(int) * (size + 1));
  91.         if(bit1 == NULL){
  92.                 cerr<<"Malloc bit1 error!"<<endl;
  93.                 exit(1);
  94.         }
  95.         memset(bit1, 0, size + 1);
  96.         setBitMask("file1", bit1);
  97.         compareBit("file2", bit1, result);
  98.         delete bit1;
  99.         cout<<result.size();
  100.         sort(result.begin(), result.end());
  101.         vector< int >::iterator        it = unique(result.begin(), result.end());
  102.         ofstream        of("result");
  103.         ostream_iterator<int>        output(of, "\n");
  104.         copy(result.begin(), it, output);
  105.         return 0;
  106. }
复制代码



这是利用位图法实现的程序,编译运行
34632_2.gif
运行时间明显比前两个少,但是这个程序是以空间换取时间,程序运行的时候分配了几百兆的空间。可见在程序设计中,方法很重要。什么情况选用什么方法。但是还是觉得前面两个方法还行,因为需要的空间比较少。



已有(1)人评论

跳转到指定楼层
pig2 发表于 2014-4-23 15:39:40
上面提到了什么位图法,这里补充一下。
位图法就是bitmap的缩写。
所谓bitmap,就是用每一位来存放某种状态,适用于大规模数据单数据状态又不是很多的情况。通常是用来判断某个数据存不存在的。
例如,要判断一千万个人的状态,每个人只有两种状态:男人,女人,可以用0,1表示。那么就可以开一个int数组,一个int有32个位,就可以表示32个人。操作的时候可以使用位操作。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

关闭

推荐上一条 /2 下一条