統計難題
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;
}