C++遍历文件夹指定扩展名

无卡OK

已于 2022-02-01 14:11:55 修改

阅读量1.1k

点赞数 2

文章标签： c++

于 2021-12-07 22:34:54 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_43901269/article/details/121780452

版权

该程序用于递归遍历指定文件夹，找出所有符合特定格式（如.txt）的文件，并将文件路径存储到vector中。使用了C++标准库中的文件操作和递归函数实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <stdio.h>
#include <vector>
#include "stdlib.h"
#include "stdio.h"
#include <string>
#include <iostream>
#include <io.h>


using namespace std;

static auto cutNext(string stream, const string &str) {
    int nPos = stream.find(str);
    if (nPos != -1) {
        stream = stream.substr(nPos + str.size(), stream.size());
    }
    return stream;
}

static auto cutPre(string stream, const string &str) {
    int nPos = stream.find(str);
    if (nPos != -1) {
        stream = stream.substr(0, nPos);
    }
    return stream;
}

void getAllFiles(string path, vector<string>& files, string format)
{
	intptr_t  hFile = 0;//文件句柄
	struct _finddata_t fileinfo;//文件信息
	string p;
	//if ((hFile = _findfirst(p.assign(path).append("/*" + format).c_str(), &fileinfo)) != -1) //文件存在
    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)//文件夹名中不含"."和".."
				{
					//files.push_back(p.assign(path).append("/").append(fileinfo.name)); //保存文件夹名
					getAllFiles(p.assign(path).append("/").append(fileinfo.name), files, format); //递归遍历文件夹
				}
			}
			else
			{
			    string x;
			    x="."+cutNext(p.assign(path).append("/").append(fileinfo.name),".");
                if (format==x||format=="")
                    files.push_back(p.assign(path).append("/").append(fileinfo.name));//如果不是文件夹，储存文件名
			}
		} while (_findnext(hFile, &fileinfo) == 0);
		_findclose(hFile);
	}
}


int main()
{
    string filePath = "文件夹路径（不要有多余的 '.' ）";
	string format = ".txt";				 //查找文件的格式
    vector<string> files;
	getAllFiles(filePath, files, format);
	int filessize = files.size();
    for (int i = 0; i < filessize; i++)
	{
		cout<<files[i].c_str()<<endl;
	}
}