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

来源：互联网发布：sql 别名后续能否使用编辑：程序博客网时间：2024/06/05 19:43

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

题意：

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

解析：

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

#include <cstdio>#include <cstring>#include <algorithm>#include <vector>#define pb push_back#define mp make_pairusing namespace std;typedef pair<int, int> 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<int> 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<int>& name = trie.val[u][0];     vector<int>& 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;}

0 0