字典樹--統計難題

統計難題

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 54150    Accepted Submission(s): 18929


 

Problem Description

Ignatius最近遇到一個難題,老師交給他很多單詞(只有小寫字母組成,不會有重複的單詞出現),現在老師要他統計出以某個字符串爲前綴的單詞數量(單詞本身也是自己的前綴).

 

 

Input

輸入數據的第一部分是一張單詞表,每行一個單詞,單詞的長度不超過10,它們代表的是老師交給Ignatius統計的單詞,一個空行代表單詞表的結束.第二部分是一連串的提問,每行一個提問,每個提問都是一個字符串.

注意:本題只有一組測試數據,處理到文件結束.

 

 

Output

對於每個提問,給出以該字符串爲前綴的單詞的數量.

 

 

Sample Input


 

banana band bee absolute acm ba b band abc

 

 

Sample Output


 

2 3 1 0

 

 

Author

Ignatius.L

 

 

Recommend

Ignatius.L   |   We have carefully selected several similar problems for you:  1075 1247 1671 1298 1800 

 

AC代碼 :

#include <cstdio>
#include <cstring>
#include <map>
#include <iostream>
using namespace std;

int main(){
    int i, len;
    char str[10];//存放輸入字符 
    map<string, int> m;
    while( gets(str) ){//輸入單詞表 
        len = strlen(str);
        if( !len )
            break;
        for(i = len; i > 0; i--){//從後往前,將所有前綴+1 
            str[i] = '\0';
            m[str]++;
        }
    }
    while( gets(str) )//輸入查詢字符串 
        printf("%d\n", m[str]);
    return 0;
}
 

真滴奇怪,字典樹模板題,用字典樹做居然超空間了,不過還是可以參考下:

#include<stdio.h>
#include<string.h>

struct node{
    int cnt;//用於記錄該前綴出現次數 
    node *next[27];//保存子節點 
    node(){
    	cnt=0;
    	memset(next,0,sizeof(next));
    }
};

void insert(node *root,char s[]){//插入單詞 
    int i=0;
    node *p=root;//創建一個指向字符串首位的node類型的指針指向root節點 
    while(s[i]){//按照字典樹的方式一位一位的插入 
        int j=s[i++]-'a';
        if(p->next[j]==NULL)//當還沒有該字母的出現,就創建一個 
			p->next[j]=new node();
        p=p->next[j]; 
        p->cnt+=1;//該前綴出現次數+1 
    }
}

int judge(node *root,char s[]){//比對 
    int i=0;
    node *p=root;
    int HowMany=0;
    while(s[i]){
        int j=s[i++]-'a';
        if(p->next[j]==NULL){
            HowMany=0;break;
        }
        p=p->next[j];
        HowMany=p->cnt;
    }
    return HowMany; 
}

int main(){
    node *root=new node();//創建一個字典樹的根節點 
    char word[10];//存放單詞表 
    char cur[10];//存放查詢前綴 
    while(gets(word)){//輸入單詞表 
        if(strcmp(word,"")==0) break;//結束標誌爲輸入爲空 
        insert(root,word);
    }
    while(gets(cur)){//輸入傲查找的字符串前綴 
        printf("%d\n",judge(root,cur));
    }
    return 0;
}

emmm,換成數組的方式來建樹,就可以AC了。。

#include<string.h>
#include<stdio.h>

int trie[400001][27],sum[400001];
int id;

void insert(char *s){
	int len=strlen(s);
	int root=0;
	for(int i=0;i<len;i++){
		int x=s[i]-'a';
		if(trie[root][x]==0)
			trie[root][x]=++id;//trie數組中放的是它的id(編號) 
		sum[trie[root][x]]++;//訪問該節點(編號)的次數sum++ 
		root=trie[root][x];//轉移到下一個節點 
	}
}

int search(char *s){
	int len=strlen(s);
	int root=0;
	for(int i=0;i<len;i++){//遍歷得出 
		int x=s[i]-'a';
		if(trie[root][x]==0)
			return 0;//不存在該節點,直接說明沒有以此的前綴 
		root=trie[root][x];//否則root = 該節點的編號 
	}
	return sum[root];//sum[i] --> 存放的是插入時訪問點i的次數 
}

int main(){
	id=0;//初始化編號 
	char s[11];
	while(gets(s)){
		if(strcmp(s,"")==0)
			break;
   		insert(s);
   	}
   	while(scanf("%s",s)!=EOF){
   		printf("%d\n",search(s));
   	}
   	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章