这道题当时用map水过了,但733ms而用trie树只需要62ms。
TRIE树
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e6+10;
char s[20];
int sum[maxn]={0};
int trie[maxn][26]={0};
int root;
int tot;
void inser()
{
root=0;
for(int i=0;s[i];i++)
{
int id=s[i]-'a';
if(!trie[root][id])
trie[root][id]=++tot; //开辟一个新的节点,节点编号是tot+1
root=trie[root][id];
sum[root]++;
}
}
void srch()
{
root=0;
for(int i=0;s[i];i++)
{
int x=s[i]-'a';
if(trie[root][x]==0) {printf("0\n"); return;}
root=trie[root][x];
}
printf("%d\n", sum[root]);
}
int main()
{
//freopen("input.txt", "r", stdin);
tot=0;
while(gets(s)&&strcmp(s, ""))
{
inser();
}
while(gets(s)!=NULL)
{
srch();
}
return 0;
}
水map代码
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <cmath>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#define mem(a,n) memset(a,n,sizeof(a))
#define FRER() freopen("input.txt", "r", stdin)
#define FREW() freopen("output.txt", "r", stdin)
typedef long long LL;
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 100000 + 10;
map<string, int> m;
int main()
{
// FRER();
string s;
string t;
while(getline(cin,s)&&s!="") {
t="";
for(int i=0;i<s.length();i++){
t+=s[i];
m[t]++;
}
}
while(cin >> s) {
printf("%d\n",m[s]);
}
return 0;
}