C++ 遍历目录文件及打印文件和其后缀

C++ 遍历目录文件及打印文件和其后缀

#include <iostream>
#include <string>
#include <windows.h>
#include <vector>
#include <assert.h>

using namespace std;

typedef std::vector<std::string> FilenameVec;


std::string GetFileExtension(std::string fileName)
{
	assert(fileName.size() > 0);
	return fileName.substr(fileName.find_last_of(".") + 1);
}


FilenameVec GetFileNameFromFolder(char *folderPath)
{
	assert(folderPath != nullptr);
	WIN32_FIND_DATA FindFileData;
	FilenameVec filenameVec;
	char dir[MAX_PATH];
	memset(dir, 0, MAX_PATH);
	sprintf_s(dir, "%s\\*.*", folderPath);

	HANDLE hFind = ::FindFirstFile(dir, &FindFileData);
	if (INVALID_HANDLE_VALUE == hFind)
	{
		filenameVec.clear();
		return filenameVec;
	}
	do
	{
		if (FindFileData.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
		{
			if (strcmp(FindFileData.cFileName, ".") == 0 ||
				st