Tirl - Trie 문자열 통계

9366 단어 데이터 구조
트 라이 문자열 통계
문자열 집합 을 유지 하고 두 가지 동작 을 지원 합 니 다.
"I x" 는 집합 에 문자열 x 를 삽입 합 니 다."Q x" 는 집합 에 문자열 이 몇 번 나 왔 는 지 물 어 봅 니 다.모두 N 개의 동작 이 있 습 니 다. 입력 한 문자열 의 총 길 이 는 105 를 넘 지 않 고 소문 자 만 포함 합 니 다.
입력 형식 첫 줄 은 정수 N 을 포함 하여 조작 수 를 표시 합 니 다.
다음 N 줄 은 각 줄 에 하나의 조작 명령 을 포함 하고 명령 은 'I x' 또는 'Q x' 중의 하나 입 니 다.
출력 형식 은 모든 질문 명령 'Q x' 에 대해 하나의 정 수 를 결과 로 출력 하여 x 가 집합 에서 나타 나 는 횟수 를 나타 낸다.
결과 마다 한 줄 을 차지한다.
데이터 범위 1 ≤ N ≤ 2 ∗ 1 0 4 1 ≤ N ≤ 2 ∗ 10 ^ 4 1 ≤ N ≤ 2 ∗ 104 5 I abc Q abc Q ab I ab Q ab 출력 사례: 1 0 1
#include 
using namespace std;
char str[10005];
int son[10005][26],cnt,con[10005];
void insert(char str[])
{
    int p=0;
    for(int i=0;str[i];i++){
        int u=str[i]-'a';
        if(!son[p][u]) son[p][u]=++cnt;
        p=son[p][u];
    }
    con[p]++;
}
int query(char str[])
{
    int p=0;
    for(int i=0;str[i];i++){
        int u=str[i]-'a';
        if(!son[p][u]) return 0;
        p=son[p][u];
    }
    return con[p];
}
int main()
{
    int n; scanf("%d",&n);
    for(int i=1;i<=n;i++){
        char op[2];
        scanf("%s",op);
        if(op[0]=='I'){
            scanf("%s",str);
            insert(str);
        }else {
            scanf("%s",str);
            printf("%d
"
,query(str)); } } }

좋은 웹페이지 즐겨찾기