hihocoder-1551-合并子目录

hihocoder-1551-合并子目录

#1551 : 合并子目录

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

小Hi的电脑的文件系统中一共有N个文件，例如：

/hihocoder/offer22/solutions/p1

/hihocoder/challenge30/p1/test  

/game/moba/dota2/uninstall  

小Hi想统计其中一共有多少个不同的子目录。上例中一共有8个不同的子目录：

/hihocoder

/hihocoder/offer22

/hihocoder/offer22/solutions

/hihocoder/challenge30

/hihocoder/challenge30/p1

/game

/game/moba

/game/moba/dota2/

输入

第一行包含一个整数N (1 ≤ N ≤ 10000)  

以下N行每行包含一个字符串，代表一个文件的绝对路径。保证路径从根目录"/"开始，并且文件名和目录名只包含小写字母和数字。  

对于80%的数据，N个文件的绝对路径长度之和不超过10000  

对于100%的数据，N个文件的绝对路径长度之和不超过500000

输出

一个整数代表不同子目录的数目。

样例输入

3  
/hihocoder/offer22/solutions/p1   
/hihocoder/challenge30/p1/test  
/game/moba/dota2/uninstall

样例输出

8

题解：

　　　知道是使用 Trie 树来解决， 但是自己做的面对 char的， 应该使用 STL，将 string 组装进 STL 进行 Trie 树的搜索才是。

AC code。

#include <unordered_map> 
#include <string> 

#include <iostream> 
using namespace std; 

struct Trie{
    unordered_map<string, Trie*> mp; 
};


int main(){
    int n, cnt; 
    while(cin >> n){
        string s; 

        Trie *root = new Trie(); 
        cnt = 0; 

        for(int i=0; i<n; ++i){
            cin >> s;
            Trie *tmp = root;  
            size_t sd = 1, fd = s.find('/', sd); 
            while(fd != string::npos){
                string t = s.substr( sd, fd - sd );  
                if(tmp->mp.find( t ) == tmp->mp.end() ){
                    tmp->mp[ t ] = new Trie(); 
                    ++cnt; 
                }
                tmp = tmp->mp[ t ]; 
                sd = fd + 1; 
                fd = s.find('/', sd); 
            } 
        } 
        cout << cnt << endl; 
    } 
    return 0; 
}

　　

My own Memory Limit Exceed Code。

我使用的是面对 ch 进行建树节点，有很多多余的节点。导致了 Memory Limit Exceeded。

#include <cstdio> 
#include <cstring> 
const int MAXN = 500000; 

struct Node{
    char ch; 
    Node *next[75];  
    Node(char _ch){
        ch = _ch;
        for(int i=0; i<75; ++i){
            next[i] = NULL; 
        } 
    }
}; 

int cnt; 
Node *root = new Node('/'); 

void insert_ch(char ch[], int len){
    Node *tmp = root; 
    for(int i=1; i<len; ++i){
        if(tmp->next[ ch[i]  - '/' ] == NULL){
            tmp->next[ ch[i] - '/' ] = new Node(ch[i]); 
            if( ch[i] == '/' ){
                cnt++; 
            } 
        }
        tmp = tmp->next[ ch[i] - '/' ]; 
    }
}

int main(){
    int n; 
    while(scanf("%d", &n) != EOF){
        cnt = 0;
        char ch[MAXN];  
        for(int i=0; i<n; ++i){
            getchar();  
            scanf("%s", ch); 
            insert_ch(ch, strlen(ch));  
        }
        printf("%d
", cnt );
    }
    return 0; 
}