文件目录结构显示与文件搜索

#include <iostream>
#include <io.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
using namespace std;

#define MaxSize 20    //next字符串的长度

typedef struct{    //匹配串结构体
string data;
int length;
}SqString;

void getString(SqString &t,string s)    //初始化匹配串结构
{
int i;
t.length = s.length();
for(i=0; i<t.length; i++)
t.data[i] = s[i];
t.data[i] = '\0';
}

void GetNext(SqString &t, int next[])    //获取next
{
int j, k;
j = 0;
k = -1;
while(j < t.length-1)
{
if(k == -1 || t.data[j] == t.data[k])
{
j++;
k++;
next[j] = k;
}
else
k = next[k];
}
}

int KMPIndex(string s, SqString &t, int next[])    //匹配相应文件夹名
{
int i = 0;
int j = 0;
while(i<s.length() && j<t.length)
{
if(j ==-1 || s[i] == t.data[j])
{
i++;
j++;
}
else
j = next[j];
}
if(j >= t.length)
return (i-t.length);//匹配到了
else
return -1;//没有匹配到
}

void searchFile(string fileName,string filePath,SqString &t,int next[])    //文件查找
{
struct _finddata_t filefind;
string curr = filePath + "\\*.*";     //修改此处改变搜索条件
int done = 0, handle;
int index;

if((handle = _findfirst(curr.c_str(), &filefind)) != -1)
{
while(!(done = _findnext(handle, &filefind)))
{
if(strcmp(filefind.name, "..") == 0)
continue;
index = KMPIndex(filefind.name, t, next);
if((_A_SUBDIR != filefind.attrib))  // 不是目录
{
if(filefind.name == fileName || index > -1)
{
cout << filePath+"\\"+filefind.name << endl;
}
}
else
{
if(filefind.name == fileName || index > -1)
{
cout << filePath+"\\"+filefind.name << endl;
}
curr = filePath + "\\" + filefind.name;
searchFile(fileName, curr, t, next);   // 递归遍历子目录
}
}
_findclose(handle);
}
}

void fileIndex(string path, int layer)  //文件目录结构输出
{
struct _finddata_t filefind;
string curr = path + "\\*.*";  // 修改此处改变搜索条件
int done = 0, i, handle;
if((handle = _findfirst(curr.c_str(), &filefind)) != -1)
{
while(!(done = _findnext(handle, &filefind)))
{
if(strcmp(filefind.name, "..") == 0)
continue;
for(i=0; i<layer; i++)
{
cout << '\t';
}
if((_A_SUBDIR == filefind.attrib))  //是目录
{
cout<<"|_*"<<filefind.name<<endl;
curr = path + "\\" + filefind.name;
fileIndex(curr, ++layer);  // 递归遍历子目录
layer--;
}
else
{
cout<<"|_"<<filefind.name<<endl;
}
}
_findclose(handle);
}
}

int main()
{
int order=0;
string filePath;  //文件路径
string fileName;  //被搜索文件名
int next[MaxSize];
next[0] = -1;
SqString s;
while(true)
{
cout << "********************************************" << endl;
cout << "**   1.文件目录输出     2.文件搜索        **" << endl;
cout << "**   0.退出                               **" << endl;
cout << "********************************************" << endl;
cout << "请输入您的指令：";
while(1)
{
cin >> order;
if(order>=0 && order<=2)
break;
printf("输入指令有误，请重新输入您的指令：");
}
switch(order)
{
case 1:
system("cls");
cout << "请输入您要搜索的文件路径（全盘搜索则输入‘0’）：";
cin >> filePath;
fileIndex(filePath, 0);////////添加写入文件  充分展示结构
break;
case 2:
system("cls");
cout << "请输入您要搜索的文件名：";
cin >> fileName;
cout << "请输入您要搜索的文件路径（全盘搜索则输入‘0’）：";
cin >> filePath;
getString(s, fileName);
GetNext(s, next);
if(filePath == "0")
{
searchFile(fileName, "C:", s, next);
searchFile(fileName, "D:", s, next);
searchFile(fileName, "E:", s, next);
}
else
{
searchFile(fileName, filePath, s, next);
}
break;
case 0:
return 0;
default:
cout << "指令输入有误，请重新输入！！！";
}
cout << "回车键继续";
getchar();
getchar();
system("cls");
}

return 0;
}

注意：1.全盘搜索部分根据个人电脑设定盘符，不知道有没有办法自动读取全盘，大神指导可留言指教。

     2.文件的输出最好存在一个多叉树中，提高代码水平

     3.文件的读取速度太慢，不要读取文件过多的文件，演示的时候时间太长

心得体会：指令防错没搞明白，没有用树形结构存储文件降低了代码的档次，所以验收的时候分值很低。不要拖延，一个星期的活干了两个月。