关键代码
void getAllFiles(string path, vector<string>& files, string fileType) {
intptr_t hFile = 0;
_finddata_t fileInfo;
string p;
if ((hFile = _findfirst(p.assign(path).append("\\*" + fileType).c_str(), &fileInfo)) != -1) {
do {
files.push_back(p.assign(path).append("\\").append(fileInfo.name));
} while (_findnext(hFile, &fileInfo) == 0);
_findclose(hFile);
}
}
#include <iostream>
#include <fstream>
#include <io.h>
#include <string>
#include <vector>
using namespace std;
int main(int argc, char **argv)
{
vector<string> files;
string extr = ".jpg";
getAllFiles(argv[1], files,extr);
for (int i = 0; i < files.size(); i++) {
cout << files[i] << endl;
}
}