问题:
给定一个文件夹,获取文件夹下的所有文件,并将文件的路径进行保存
解决方法:
采用递归的思想,在递归过程中对每一层进行遍历。若为文件夹,继续向下遍历。若为文件,将路径保存。
实现代码:
#include <iostream>
#include <fstream>
#include <io.h>
#include <string>
#include <vector>
using namespace std;
void getAllFiles( string path, vector<string>& files)
{
long hFile = 0; //文件句柄
struct _finddata_t fileinfo; //文件信息读取结构
string p;
if((hFile = _findfirst(p.assign(path).append("\\*").c_str(),&fileinfo)) != -1)
{
do
{
if((fileinfo.attrib & _A_SUBDIR)) //比较文件类型是否是文件夹
{
if(strcmp(fileinfo.name,".") != 0 && strcmp(fileinfo.name,"..") != 0)
{
getAllFiles( p.assign(path).append("\\").append(fileinfo.name), files ); //如果是文件夹,继续向下遍历
}
}
else //是文件
{ files.push_back(p.assign(path).append("\\").append(fileinfo.name) ); //是文件,将其路径加入到files集合中
}
}while(_findnext(hFile, &fileinfo) == 0); //寻找下一个,成功返回0,否则-1
_findclose(hFile);
}
}
int main()
{
char * filePath = "file"; //所要查找的文件夹
char * distAll = "path.txt"; //结果保存
vector<string> files; //保存文件路径信息
ofstream ofn(distAll); //打开文件
int size = 0; //存储文件个数
while(!files.empty()) //清空vector
{
files.pop_back();
}
getAllFiles(filePath, files); //递归查找文件以及文件夹。文件夹路径为filePath
size = files.size(); //包含文件个数
cout << size << endl;
for (int i = 0; i < size; i++)
{
ofn << files[i] << endl; //把文件路径保存
}
ofn.close(); //文件关闭
return 0;
}