2025CCPC郑州-E.双生魔咒个人题解

E.双生魔咒

#字符串 #trie

题目

思路

从感性上理解，本题肯定是在相似度尽可能大的前提下对半分最优。

那么在对半分之后，如何计算出答案呢？

这里想到了通过构建一棵trie来维护每一种前缀的出现次数，根据每一种前缀的出现次数来算当前前缀对最后答案的贡献

拿样例举例子：
比如对于前缀w，一共有四个单词经过这个节点：why,whom,well,weels，那么这个节点的cnt就是4
首先要理解一件事情：比如单词ennaimez与ennus对答案的总贡献为3，那么可以理解为trie上的节点e贡献1、节点n贡献1、节点n贡献1,如图：

那么，通过trie得到每个节点的cnt值之后，怎么计算出当前节点对最后答案的贡献呢？
用图中的第一个节点e来理解，其cnt值为2，代表有两个字符串经过这个节点
这两个字符串分别分到根和引两边进行匹配，匹配后对答案的贡献为1（因为这只是一个字母）。
那么如果当前节点的cnt值为3呢？
那么代表有三个字符串经过这个节点，这三个串可以左边分一个右边分两个，一共有2对匹配，贡献为2。
因此总结出规律：
已知当前节点的cnt值，代表有cnt个串经过当前节点。可以将这cnt个串左右平分：左边cnt/2个，右边cnt-cnt/2个，那么左右两两匹配的个数就是做乘法cnt/2*(cnt-cnt/2)
因此，将整棵trie的所有节点遍历一遍，贡献加起来就可以啦~

代码实现

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define rep(i,a,b) for(ll i=(a);i<=(b);i++)
#define per(i,a,b) for(ll i=(a);i>=(b);i--)
const ll inf = 0x3f3f3f3f3f3f3f3f;

const ll Ns = 1e5 + 5;
int ch[Ns][100], cnt[Ns], idx;

int getnum(char s) {
    if (s - '0' < 10) { return s - '0'; }
    else { return s - 'A' + 10; }
}
void insert(string& s) {
    int len = s.length() - 1, p = 0;
    rep(i, 0, len) {
        int j = getnum(s[i]);
        if (!ch[p][j])ch[p][j] = ++idx;
        p = ch[p][j];
        cnt[p]++;
    }
}

void eachT() {
    ll n; cin >> n;
    n *= 2;
    rep(i, 1, n) {
        string s; cin >> s;
        insert(s);
    }
    ll ans = 0;
    rep(i, 0, idx) {
        ans += cnt[i] / 2 * (cnt[i] - cnt[i] / 2);
    }
    cout << ans;
}
int main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    ll t = 1;
    //cin >> t;
    while (t--) { eachT(); }
}