ACWING 835. Trie字符串统计 (字典树)

https://www.acwing.com/problem/content/837/

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

const int N=100010;
int son[N][26],cnt[N];
int idx;

void insert(char str[]) {
	int p=0;
	int u;
	for(int i=0;str[i];i++) {
		u=str[i]-'a';
		if(!son[p][u]) son[p][u]=++idx;
		p=son[p][u];
	}
	cnt[p]++;
}

int query(char str[]) {
	int p=0;
	int u;
	for(int i=0;str[i];i++) {
		u=str[i]-'a';
		if(!son[p][u]) return 0;
		p=son[p][u];
	}
	return cnt[p];
}

int main() {

	int n;
	cin>>n;
	char op;
	char str[N];
	while(n--) {
		cin>>op>>str;
		if(op=='I') insert(str);
		else cout<<query(str)<<endl;
	}


	return 0;
}

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

const int N=100010;
int son[N][2],cnt[N],idx;

int toBinary(int a[],int x) {
    int len=0;
    do  if(x&1) a[len++]=1;
        else a[len++]=0;
    while(x>>=1);
    return len;
}

void insert(int a[], int len) {
    int q=0;

//    cout<<"insert\n";
//    for(int i=len-1; i>=0; i--) cout<<a[i];
//    cout<<endl;

    for(int i=len-1; i>=0; i--) {
//		cout<<"son[q][a[i]] = "<<<<"\nq = "<<q<<endl;
        if(!son[q][a[i]]) son[q][a[i]]=++idx;
//        printf("son[%d][%d] = %d\n",q,a[i],son[q][a[i]]);
        q=son[q][a[i]];
    }
//    cout<<"endl\n";
    cnt[q]++;
}

int query(int x) {
	int a[32];
	int len=toBinary(a,x);

	int q=0;
	for(int i=len-1;i>=0;i--) {
//		printf("p = %d, a[i] = %d, son[q][a[i]] = %d\n",q,a[i],son[q][a[i]]);
		if(!son[q][a[i]]) return 0;
		q=son[q][a[i]];
	}
	return cnt[q];
}

int main() {

    int n;
    cin>>n;
    int a[32],x;
    int len;
    for(int i=0; i<n; i++) {
        cin>>x;
        len=toBinary(a,x);

//        cout<<"x = "<<x<<"\n";
//        for(int i=len-1; i>=0; i--) cout<<a[i];
//        cout<<endl;

        insert(a,len);
    }

    for(int i=0; i<idx; i++) {
        for(int j=0; j<2; j++) {
            if(son[i][j]>9) printf("|%d |",son[i][j]);
            else printf("| %d |",son[i][j]);
        }
        cout<<endl;
    }
    cout<<endl;

//    for(int i=0;i<idx*2;i++) cout<<cnt[i]<<" ";
//    cout<<endl;

	int m;
	cin>>m;
	int xm;
	while(m--) {
		cin>>xm;
		cout<<query(xm)<<endl;
	}

    return 0;
}

/*

3
1 2 6

*/

/*

写着玩

*/