lightoj 1114(trie)

题意:给出n个单词,其中首尾字母相同,中间字母任意排列的单词是同一个码,然后给出m个句子,每个句子含有若干个单词,问多少个不同的句子和这个句子有相同的编码。
题解:把每个单词除去首尾的部分按字典序排序,然后插入字典树,val[u]存节点出现次数,在每个单词的最后一个节点数量累加,然后把输入的句子中每个统计到的单词数量乘起来。
用自己写的句子按空格拆分单词的代码一直wa,看了题解用的strtok(str, ” “)函数,可以把一个字符串按分隔符‘ ’拆分成为一组字符串,用这个函数就可以a,我也是醉了。。。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 130005;
int Next[N][60], val[N], sz, n, m;
char str[10005];

void init() {
    memset(Next[0], 0, sizeof(Next[0]));
    val[0] = 0;
    sz = 1;
}

void insert(char *s, int len) {
    int u = 0;
    for (int i = 0; i < len; i++) {
        int k;
        if (s[i] >= 'a' && s[i] <= 'z')
            k = s[i] - 'a' + 1;
        else
            k = s[i] - 'A' + 27;
        if (!Next[u][k]) {
            memset(Next[sz], 0, sizeof(Next[sz]));
            val[sz] = 0;
            Next[u][k] = sz++;
        }
        u = Next[u][k];
    }
    val[u]++;
}

int query(char *s, int len) {
    int u = 0;
    for (int i = 0; i < len; i++) {
        int k;
        if (s[i] >= 'a' && s[i] <= 'z')
            k = s[i] - 'a' + 1;
        else
            k = s[i] - 'A' + 27;
        if (!Next[u][k])
            return 0;
        u = Next[u][k];
    }
    return val[u];
}

int main() {
    int t, cas = 1;
    scanf("%d", &t);
    while (t--) {
        init();
        scanf("%d", &n);
        for (int i = 0; i < n; i++) {
            scanf("%s", str);
            int len = strlen(str);
            if (len > 2)
                sort(str + 1, str + len - 1);
            insert(str, len);
        }
        scanf("%d", &m);
        getchar();
        printf("Case %d:\n", cas++);
        for (int i = 0; i < m; i++) {
            long long res = 1;
            gets(str);
            char *p = strtok(str, " ");
            while (p) {
                char s[10005];
                strcpy(s, p);
                int len = strlen(s);
                if (len > 2)
                    sort(s + 1, s + len - 1);
                res *= query(s, len);
                p = strtok(NULL, " ");
            }
            printf("%lld\n", res);
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值