今天我遇到了这样一个任务：要求编写一个程序，统计和这个程序在同一目录下（及其子目录）所有文件的单词数。统计单词数十分倒不是太难，倒是找出同一目录下的所有文件，是我从来没有接触过的。仔细分析，这个问题其实包含两个小问题：
1. 如何获取当前程序所在文件夹的路径
2. 如何给定一个路径，递归地找到其中（包括子目录）所有文件的文件名（相对路径）
那我们就依次解决这两个问题。
.
.
1. 如何获取当前程序所在文件夹的路径
.
.
解决这个问题只需要一个简单的函数即可（需要包含一个输入输出库）：
#include <io.h>
char *getcwd( char *buffer, int maxlen );
这个函数能够获取当前的工作目录，具体来说，它会将当前工作目录的绝对路径复制到参数buffer所指的内存空间中,参数maxlen为buffer的空间大小。
.
我们可以写一个程序来测试一下。
#include<iostream>
#include<io.h>
using namespace std;
#define MAX_PATH 80
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);
它的用处是，给定一个文件目录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);
}
}
.
下面给出一个运用这个函数的例子：
#include<iostream>
#include<vector>
#include<io.h>
using namespace std;
#define MAX_PATH 80
void getFiles( string path, vector<string>& files );
int main()
{
vector<string> files;
char * filePath = "C:\\Users\\Star\\Desktop\\SoftTest";
////获取该路径下的所有文件
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;
#define MAX_PATH 80
void getFiles( string path, vector<string>& files );
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;
#define MAX_PATH 80
void getFiles( string path, string path2, vector<string>& files );
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);
}
}
.
运行结果：
成功地给出了相对路径！！
.
.
也可以给出文件名：
.
.
.