今天我遇到了这样一个任务：要求编写一个程序，统计和这个程序在同一目录下（及其子目录）所有文件的单词数。统计单词数十分倒不是太难，倒是找出同一目录下的所有文件，是我从来没有接触过的。仔细分析，这个问题其实包含两个小问题：
1. 如何获取当前程序所在文件夹的路径
2. 如何给定一个路径，递归地找到其中（包括子目录）所有文件的文件名（相对路径）
那我们就依次解决这两个问题。
.
.
1. 如何获取当前程序所在文件夹的路径
.
.
解决这个问题只需要一个简单的函数即可（需要包含一个输入输出库）：
#include <io.h>
char *getcwd( char *buffer, int maxlen );- 1
- 2
- 1
这个函数能够获取当前的工作目录，具体来说，它会将当前工作目录的绝对路径复制到参数buffer所指的内存空间中,参数maxlen为buffer的空间大小。
.
我们可以写一个程序来测试一下。
#include<iostream>
#include<io.h>
using namespace std;
- 1
- 2
- 3
- 4
- 1
- 2
- 3
int main()
{
char buffer[MAX_PATH];
getcwd(buffer, MAX_PATH);
cout << buffer << endl;
return 0;
}
.
运行成功！
.
.
.
2. 如何给定一个路径，递归地找到其中（包括子目录）所有文件的文件名（相对路径）
.
.
这里我使用了网上现成的代码，先附上链接。
.
这里，他定义了一个函数（需要包含一个向量库，当然，还有我们在上一个问题里需要使用的输入输出库）
#include <io.h>
#include<vector>
void getFiles(string path, vector<string>& files);- 1
- 2
- 3
- 1
- 2
它的用处是，给定一个文件目录path，然后找到其中所有（包括其子目录里）的文件，将文件名放入files中，files是一个字符串向量的引用类型。
.
具体的函数内容是这样的：
void getFiles( 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)
getFiles( p.assign(path).append("\").append(fileinfo.name), files );
}
else
{
files.push_back(p.assign(path).append("\").append(fileinfo.name) );
}
}while(_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
.
下面给出一个运用这个函数的例子：
#include<iostream>
#include<vector>
#include<io.h>
using namespace std;
- 1
- 2
- 3
- 4
- 5
- 1
- 2
- 3
- 4
int main()
{
vector<string> files;
char * filePath = “C:UsersStarDesktopSoftTest”;
获取该路径下的所有文件
getFiles(filePath, files);
char str[30];
int size = files.size();
for (int i = 0;i < size;i++)
{
cout<<files[i].c_str()<<endl;
}
return 0;
}
运行成功！
.
.
.
3. 组合以后的完整代码
.
#include<iostream>
#include<vector>
#include<io.h>
using namespace std;
- 1
- 2
- 3
- 4
- 5
- 1
- 2
- 3
- 4
int main()
{
vector<string> files;
char buffer[MAX_PATH];
getcwd(buffer, MAX_PATH);
char * filePath = buffer;
获取该路径下的所有文件
getFiles(filePath, files );
char str[30];
int size = files.size();
for (int i = 0;i < size;i++)
{
cout<<files[i].c_str()<<endl;
}
return 0;
}
void getFiles( 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)
getFiles( p.assign(path).append("").append(fileinfo.name), files );
}
else
{
files.push_back(p.assign(path).append("").append(fileinfo.name) );
}
}while(_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}
.
这段代码的运行结果和上面的运行结果相同，我就不给出运行结果了。
.
.
.
4.返回相对路径
.
之前的那种方法很好，不过唯一的缺点是，函数返回的都是绝对路径，如果想要改成相对路径，该怎么办呢？
.
.
我将代码进行了修改，具体做法是在getFile()函数中加入了一个path2参数，改进后的代码如下：
.
#include<iostream>
#include<vector>
#include<io.h>
using namespace std;
- 1
- 2
- 3
- 4
- 5
- 1
- 2
- 3
- 4
int main(){
vector<string> files;
char buffer[MAX_PATH];
getcwd(buffer, MAX_PATH);
string filePath;
filePath.assign(buffer).append("");
获取该路径下的所有文件
getFiles(filePath,"", files );
char str[30];
int size = files.size();
for (int i = 0;i < size;i++)
{
cout<<files[i].c_str()<<endl;
}
return 0;
}
void getFiles( string path, string path2, vector<string>& files )
{
//文件句柄
long hFile = 0;
//文件信息
struct _finddata_t fileinfo;
string p,p2;
if((hFile = _findfirst(p.assign(path).append(path2).append("*").c_str(),&fileinfo)) != -1)
{
do
{
//如果是目录,迭代之
//如果不是,加入列表
if((fileinfo.attrib & _A_SUBDIR))
{
if(strcmp(fileinfo.name,".") != 0 && strcmp(fileinfo.name,"…") != 0)
getFiles( p.assign(path).append(""),p2.assign(fileinfo.name).append(""), files );
}
else
{
files.push_back(p.assign(path2).append(fileinfo.name) ); //这一行可以给出相对路径
//files.push_back(p.assign(fileinfo.name) ); //这一行可以给出文件名
}
}while(_findnext(hFile, &fileinfo) == 0);
_findclose(hFile);
}
}
.
运行结果：
成功地给出了相对路径！！
.
.
也可以给出文件名：
.
.
.