HDOJ 3460 Ancient Printer (字典树 + 深度遍历)

来源:互联网 发布:mac excel好用吗 编辑:程序博客网 时间:2024/05/29 13:20


点击打开链接


这道题在字典树建立的时候我不再是套用模板,而是更加自然地表达了自己的想法。

另外我更深刻地理解了字典树。trie只是定义了字典树最小的一个单元的类型,tree[]是最小单元,即一个节点加上它从它延伸出来的箭头,也许还会包括val这样的附加值。但是一棵完整的字典树,必须是由这一些最小单元,以某种特殊方式结合起来而组成的。insert()就是定义了这样一种特殊方式,称为建树。

这道题仔细思考一下,我发现如果把所有的单词打一遍再删掉,最小的操作数就是深度遍历一遍的步数,不过回溯时步数也要算上。但是最后一个单词是不用删掉的,那么这个单词如果是最长的那个,最合算。但是是不是一定能把最长的那个单词安排到最后一个呢? 如果把前缀相同的单词打包在一起,不同的包之间交换没有问题,同一个包直接不同单词交换也是没有问题的(深度遍历时先选择哪一个分支没有关系)。所以一定可以安排到最后一个。每一个单词必定要print一次。

综上,最后的答案就是ans - max + n.

#include<cstdio>#include<cstring>#include<string>#include<iostream>#define cg(x) x - 'a';using namespace std;const int MAXN = 1e6;struct trie{int next[26];int val;}tree[MAXN];int ans = 0, sum = 0;void insert(string s){    int root = 0;    int l = s.size();    for (int i = 0; i < l; i++)    {        int c = cg(s[i]);        if (tree[root].next[c]) root = tree[root].next[c];        else        {            sum++;            tree[root].next[c] = sum;            root = sum;            memset(&tree[root], 0, sizeof(trie));        }        tree[root].val++;    }}void find(int root){    for (int i = 0; i < 26; i++)        if (tree[root].next[i] != 0)        {            ans++;            find(tree[root].next[i]);            ans++;        }}int main(){    int n;    while (cin >> n)    {        string s;        ans = 0;        memset(&tree[0], 0, sizeof(trie));        int max = 0;        for (int i = 1; i <= n; i++)        {            cin >> s;            insert(s);            int l = s.size();            if (l > max) max = l;        }        find(0);        cout << ans - max + n << endl;    }    return 0;}

0 0