[Trie树] 文件夹计数

题目描述

  可以用路径来描述文件夹,路径为一个包含若干部分的字符串,之间用’/’分隔。每部分均为一个文件夹的名称,且表示这个文件夹的父文件夹为前一部分描述的文件夹。
  例如:/home/fj/summer表示根目录下有一个名称为home的文件夹,这个home文件夹下有一个名称fj的文件夹,这个名称为fj的文件夹下有。
  每个路径的第1个字符总是’/’,且不会出现两个连续的’/’,最后一个字符不会是’/’。而所有文件夹仅包含数字和字母。
  现在先给出N个路径,一开始除了根目录不存在任何文件夹,在每给出一个路径后,对于第i个路径,你需要输出的是若要让第1个路径到第i个路径存在,最少需要新建多少个文件夹。

输入格式

输入文件第1行为一个正整数N。
接下来N行,每行为一个描述路径的字符串,长度均不超过100。

输出格式

输出应包含N行,每行1个正整数,第i行输出若要使第1个路径到第i个路径存在,最少需要新建多少个文件夹。

样例数据

样例输入

样例数据#1
2
/home/fj/summer
/home/fj/123
样例数据#2
3
/chicken
/chicken/egg
/chicken
样例数据#3
4
/a
/a/b
/a/c
/b/b

样例输出

样例数据#1
3
4
样例数据#2
1
2
2
样例数据#3
1
2
3
5

数据规模

对于所有数据,N<=1000。
对于部分数据,有N<=20;
对于部分数据,有N<=200;
对于部分数据,有对于所有路径最多存在两个’/’(包含第1个字符)。

题目分析

对于每一个路径,对’/’中间的字符串累加放在trie树里,当然最后要添加一个’/’,最后统计单词个数即可,具体看代码

源代码

#include<algorithm>
#include<iostream>
#include<iomanip>
#include<cstring>
#include<cstdlib>
#include<vector>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
inline const int Get_Int() {
    int num=0,bj=1;
    char x=getchar();
    while(x<'0'||x>'9') {
        if(x=='-')bj=-1;
        x=getchar();
    }
    while(x>='0'&&x<='9') {
        num=num*10+x-'0';
        x=getchar();
    }
    return num*bj;
}
const int maxn=300000;
struct Tree {
    int child[65],bj;
    void clear() {
        memset(child,0,sizeof(child));
        bj=0;
    }
};
struct Trie {
    Tree tree[maxn];
    int cnt,Word;
    void init() {
        Word=0;
        cnt=0;
        memset(tree,0,sizeof(tree));
    }
    void insert(string s) { //插入字符串s 
        int x=0;
        for(int i=0; i<s.length(); i++) {
            int y;
            if(s[i]>='A'&&s[i]<='Z')y=s[i]-'A';
            else if(s[i]>='a'&&s[i]<='z')y=s[i]-'a'+26;
            else if(s[i]>='0'&&s[i]<='9')y=s[i]-'0'+52;
            else y=63;
            if(tree[x].child[y]==0) { //没有此结点,新建结点 
                tree[x].child[y]=++cnt;
                tree[cnt].clear();
                x=cnt;
            } else x=tree[x].child[y];
        }
        tree[x].bj=1;
        Word++;
    }
    int Find(string s) { //返回字符串s出现的下标,没有返回-1 
        int x=0;
        for(int i=0; i<s.length(); i++) {
            int y;
            if(s[i]>='A'&&s[i]<='Z')y=s[i]-'A';
            else if(s[i]>='a'&&s[i]<='z')y=s[i]-'a'+26;
            else if(s[i]>='0'&&s[i]<='9')y=s[i]-'0'+52;
            else y=63;
            if(tree[x].child[y]==0)return -1; //不在树中
            x=tree[x].child[y];
        }
        if(tree[x].bj)return x; //有标记域才能返回 
        else return -1;
    }
};
Trie trie;
int n;
int main() {
    ios::sync_with_stdio(false);
    cin>>n;
    trie.init();
    for(int i=1; i<=n; i++) {
        string s,tmp="";
        cin>>s;
        int len=s.length();
        s=' '+s;
        for(int i=2; i<=len; i++)
            if(s[i]=='/') {
                tmp+='/';
                if(trie.Find(tmp)==-1)trie.insert(tmp);
            } else tmp+=s[i];
        tmp+='/';
        if(trie.Find(tmp)==-1)trie.insert(tmp);
        printf("%d\n",trie.Word);
    }
    return 0;
}
    原文作者:Trie树
    原文地址: https://blog.csdn.net/Bill_Yang_2016/article/details/54698087
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞