广度优先遍历文件夹和深度递归遍历文件夹

// 广度遍历目录

BOOL TraverseDirBF(const string& strDir)
{
    deque<string> dequePath;
    string strTemp = strDir;

    int iFileCount = 0;
    int iDirCount = 0;

    ofstream ofs("out.txt");

    while(TRUE)
    {
        if (!PathFileExists(strTemp.c_str()))
        {
            return FALSE;
        }

        string strDirFind = strTemp + string("\\*");
        WIN32_FIND_DATA ffd = {0};
        HANDLE hFind = FindFirstFile(strDirFind.c_str(), &ffd);

        if (INVALID_HANDLE_VALUE == hFind)
        {
            //return FALSE;
            if (dequePath.empty())
            {
                break;
            }
            else
            {
                strTemp = dequePath.front();
                dequePath.pop_front();
                continue;
            }
        }

        // List all the files in the directory with some info about them.

        do
        {
            if ((0 == strcmp(ffd.cFileName, ".")) || (0 == strcmp(ffd.cFileName, "..")))
            {
                continue;
            }

            string strFilePathName = strTemp + "\\" + string(ffd.cFileName);

            if (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
            {
                //_tprintf(TEXT("  %s   <DIR>\n"), strFilePathName.c_str());
                string str111 = strFilePathName + "\n";
                ofs.write(str111.c_str(), str111.size());
                dequePath.push_back(strFilePathName);
                ++iDirCount;
            }
            else
            {
                //_tprintf(TEXT("  %s   <FILE>\n"), strFilePathName.c_str());
                string str111 = strFilePathName + "\n";
                ofs.write(str111.c_str(), str111.size());
                ++iFileCount;

            }
        }
        while (FindNextFile(hFind, &ffd) != 0);

        DWORD dwError = GetLastError();
        if (dwError != ERROR_NO_MORE_FILES)
        {
            FindClose(hFind);
            return FALSE;
        }

        FindClose(hFind);
        if (dequePath.empty())
        {
            break;
        }
        else
        {
            strTemp = dequePath.front();
            dequePath.pop_front();
        }
    }
    ofs.close();
    return TRUE;
}

 

 

 

// 深度优先遍历
BOOL TraverseDirDF(const string& strDir, int& iFileCount, int& iDirCount)
{
    if (!PathFileExists(strDir.c_str()))
    {
        return FALSE;
    }

    string strDirFind = strDir + string("\\*");
    WIN32_FIND_DATA ffd = {0};
    HANDLE hFind = FindFirstFile(strDirFind.c_str(), &ffd);
    if (INVALID_HANDLE_VALUE == hFind)
    {
        return FALSE;
    }

    do
    {
        if ((0 == strcmp(ffd.cFileName, ".")) || (0 == strcmp(ffd.cFileName, "..")))
        {
            continue;
        }

        string strFilePathName = strDir + "\\" + string(ffd.cFileName);

        if (ffd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
        {
            ++iDirCount;
            TraverseDirDF(strFilePathName, iFileCount, iDirCount);
        }
        else
        {
            ++iFileCount;
        }
    }
    while (FindNextFile(hFind, &ffd) != 0);

    DWORD dwError = GetLastError();
    if (dwError != ERROR_NO_MORE_FILES)
    {
        FindClose(hFind);
        return FALSE;
    }

    FindClose(hFind);
    return TRUE;
}

 

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值