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

点击打开链接

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

另外我更深刻地理解了字典树。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;
}