LA4122

哈夫曼树+搜索

抄了抄代码

先开始不知道怎么限制哈夫曼树，然后看了看代码，是用bfs序来限制。因为每个节点的右子树节点肯定不小于左儿子，同一层也是。所以先搞出bfs序，然后搜索，判断每一层右边是否大于左边。

哈夫曼树的每个节点必然会有零个或两个儿子，这也是判断无解或有解的情况。

#include<bits/stdc++.h>
using namespace std;
const int N = ;
string s;
int n, cnt, ans, root, now, kase;
int child[N][], val[N], pos[N];
vector<int> order;
void insert(int x)
{
    for(int i = ; i < s.length(); ++i)
    {
        int t = s[i] - '';
        if(!child[x][t])
            child[x][t] = ++cnt;
        x = child[x][t];
    }
}
void dfs(int u)
{
    if(pos[u] == order.size() - )
    {
        ++ans;
        return;
    }
    if(child[u][] == )
    {
        dfs(order[pos[u] + ]);
        return;
    }
    for(int i = ; i *  <= val[u]; ++i)
    {
        val[child[u][]] = i;
        val[child[u][]] = val[u] - i;
        if(val[order[pos[child[u][]] - ]] && val[order[pos[child[u][]] - ]] < val[child[u][]]) continue;
        dfs(order[pos[u] + ]);
    }
    val[child[u][]] = val[child[u][]] = ;
}
bool build()
{
    for(int i = ; i <= cnt; ++i) if(child[i][] * child[i][] ==  && child[i][] + child[i][])
        return ;
    queue<int> q;
    order.clear();
    q.push();
    while(!q.empty())
    {
        int u = q.front();
        q.pop();
        pos[u] = order.size();
        order.push_back(u);
        if(child[u][]) q.push(child[u][]);
        if(child[u][]) q.push(child[u][]);
    }
    return true;
}
int main()
{
    while(cin >> n)
    {
        if(!n)
            break;
        memset(child, , sizeof(child));
        ans = ;
        cnt = ;
        root = ++cnt;
        for(int i = ; i <= n; ++i)
        {
            cin >> s;
            insert(root);
        }
        printf("Case %d: ", ++kase);
        if(!build())
        {
            puts("");
            continue;
        }
        val[] = ;
        dfs();
        cout << ans << endl;
    }
    return ;
}