多对多LCP和最大 Trie DFS CodeForces - 566A Matching Names

http://codeforces.com/problemset/problem/566/A

题意：

有n个学生在学校，他们有n个真名，以及n个假名。 
求如何真名和假名，匹配使得LCP和最大。

解析：

先给真名和假名标号，然后插入到字典树上。 
一颗字典树上面的每个节点，保存的是每个字符串前缀的编号。 
然后对字典树进行dfs，然后优先选择深的匹配，并标记。 
然后回溯匹配，被标记过的不要。

#include 
#include 
#include 
#include 
#define pb push_back
#define mp make_pair
using namespace std;
typedef pair pii;
const int MAXN = 1e5 + 10;
const int maxnode = 800005;
const int sigma_size = 26;

vector< pii > ans; 

struct Trie {
    int ch[maxnode][sigma_size];
    vector val[maxnode][2];
    int sz;
    void clear() {sz = 1; memset(ch[0], 0, sizeof(ch[0]));}
    Trie() { clear(); }
    int idx(int c) { return c - 'a'; }

    void insert(char *s, int id, int v) {
        int u = 0, n = strlen(s);
        val[u][id].pb(v);
        for(int i = 0; i < n; i++) {
            int c = idx(s[i]);
            if(!ch[u][c]) {
                memset(ch[sz], 0, sizeof(ch[sz]));
                val[sz][id].clear();
                ch[u][c] = sz++;
            }
            u = ch[u][c];
            val[u][id].pb(v);
        }
    }
} trie;

bool vis[2][MAXN];
int n, total;

void init() {
    trie.clear();
    ans.clear();
    memset(vis, false, sizeof(vis));
}

void dfs(int u, int dep) {
    for(int i = 0; i < sigma_size; i++) {
        if(trie.ch[u][i]) {
            dfs(trie.ch[u][i], dep+1);
        }
    }
    vector& name = trie.val[u][0]; 
    vector& pse = trie.val[u][1];
    int a, b;
    for(int i = 0; i < name.size(); i++) {
        a = name[i];
        if(vis[0][a]) continue;
        for(int j = 0; j < pse.size(); j++) {
            b = pse[j];
            if(vis[1][b]) continue;
            vis[0][a] = vis[1][b] = true;
            ans.pb(mp(a, b));
            total += dep;
            break;
        }
    }
}

char str[maxnode];
int main() {
    while(scanf("%d", &n) != EOF) {
        init();
        for(int i = 1; i <= n; i++) {
            scanf("%s", str);
            trie.insert(str, 0, i);
        }
        for(int i = 1; i <= n; i++) {
            scanf("%s", str);
            trie.insert(str, 1, i);
        }
        total = 0;
        dfs(0, 0);
        printf("%d\n", total);
        for(int i = 0; i < ans.size(); i++) {
            printf("%d %d\n", ans[i].first, ans[i].second);
        }
    }
    return 0;
}