c++遍历目录下所有子目录及文件

1. #include <io.h>  
2. #include <fstream>  
3. #include <string>  
4. #include <vector>  
5. #include <iostream>  
6. using namespace std;  
7.   
8.   //其实两个字符串连在一起比如string可惜写成：str1+str2+“xxx”。。。
9. //获取所有的文件名  
10. void GetAllFiles( string path, vector<string>& files)    
11. {    
12.   
13.     long   hFile   =   0;    
14.     //文件信息    
15.     struct _finddata_t fileinfo;//用来存储文件信息的结构体    
16.     string p;    
17.     if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) !=  -1)  //第一次查找  
18.     {    
19.         do    
20.         {     
21.             if((fileinfo.attrib &  _A_SUBDIR))  //如果查找到的是文件夹  
22.             {    
23.                 if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)  //进入文件夹查找  
24.                 {  
25.                     files.push_back(p.assign(path).append("\\").append(fileinfo.name) );  
26.                     GetAllFiles( p.assign(path).append("\\").append(fileinfo.name), files );   
27.                 }  
28.             }    
29.             else //如果查找到的不是是文件夹   
30.             {    
31.                 files.push_back(p.assign(fileinfo.name) );  //将文件路径保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)  
32.             }   
33.   
34.         }while(_findnext(hFile, &fileinfo)  == 0);    
35.   
36.         _findclose(hFile); //结束查找  
37.     }   
38.   
39. }    
40.   
41. //获取特定格式的文件名  
42. void GetAllFormatFiles( string path, vector<string>& files,string format)    
43. {    
44.     //文件句柄    
45.     long   hFile   =   0;    
46.     //文件信息    
47.     struct _finddata_t fileinfo;    
48.     string p;    
49.     if((hFile = _findfirst(p.assign(path).append("\\*" + format).c_str(),&fileinfo)) !=  -1)    
50.     {    
51.         do    
52.         {      
53.             if((fileinfo.attrib &  _A_SUBDIR))    
54.             {    
55.                 if(strcmp(fileinfo.name,".") != 0  &&  strcmp(fileinfo.name,"..") != 0)    
56.                 {  
57.                     //files.push_back(p.assign(path).append("\\").append(fileinfo.name) );  
58.                     GetAllFormatFiles( p.assign(path).append("\\").append(fileinfo.name), files,format);   
59.                 }  
60.             }    
61.             else    
62.             {    
63.                 files.push_back( p.assign(fileinfo.name));  //将文件路径保存，也可以只保存文件名:    p.assign(path).append("\\").append(fileinfo.name)  
64.             }    
65.         }while(_findnext(hFile, &fileinfo)  == 0);    
66.   
67.         _findclose(hFile);   
68.     }   
69. }   
70.   
71. // 该函数有两个参数，第一个为路径字符串(string类型，最好为绝对路径)；  
72. // 第二个参数为文件夹与文件名称存储变量(vector类型,引用传递)。  
73. // 在主函数中调用格式(并将结果保存在文件"AllFiles.txt"中，第一行为总数)：  
74.   
75. int main()  
76. {  
77.     string filePath = "D:\\文档\\课程\\数据结构\\MOOC练习\\testPCL\\x64\\Debug";    
78.     vector<string> files;    
79.     char * distAll = "AllFiles.txt";  
80.   
81.     //读取所有的文件，包括子文件的文件  
82.     //GetAllFiles(filePath, files);  
83.   
84.     //读取所有格式为jpg的文件  
85.     string format = ".dll";  
86.     GetAllFormatFiles(filePath, files,format);  
87.     ofstream ofn(distAll);  
88.     int size = files.size();   
89.     ofn<<size<<endl;  
90.     for (int i = 0;i<size;i++)    
91.     {    
92.         ofn<<files[i]<<endl; // 写入文件  
93.         cout<< files[i] << endl;//输出到屏幕  
94.     }  
95.     ofn.close();  
96.     return 0;  
97. }