hdu 1251字典樹模板

 一個查詢前綴，很明顯的字典樹

關鍵是空行的處理。 我很好奇爲什麼fgets會報RE 用gtes就不會。但是現在C++14 gets都刪了，編譯不了。emmm

#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#include <stdio.h>
#include <ctype.h>
#include <bitset>
#define  LL long long
#define  ULL unsigned long long
#define mod 10007
#define INF 0x7ffffff
#define mem(a,b) memset(a,b,sizeof(a))
#define MODD(a,b) (((a%b)+b)%b)
using namespace std;
const int maxn = 2e6 + 5;
int n,m;
int flag;
int str[maxn][30];
int sum[maxn];
char ss[maxn];
void Insert(char *ss)
{
    int p = 0;
    for(int i = 0; i < strlen(ss); i++){
      int c = ss[i] - 'a';
      if(!str[p][c]) str[p][c] = ++flag;
      sum[str[p][c]]++;
      p = str[p][c];
    }
}
int query(char *k)
{
    int p = 0;
    for(int i = 0; i < strlen(k); i++){
      int c = k[i] - 'a';
      if(!str[p][c]) return 0;
      p = str[p][c];
    }
    return sum[p];
}
int main()
{
    flag = 0;
    while(gets(ss)){
        if(ss[0] == '\0') break;
        Insert(ss);
    }
    char k[maxn];
    //printf("111");
    while(~scanf("%s",k)){
        printf("%d\n",query(k));
    }

    return 0;
}