公司命名——计算交集

题解：2306. 公司命名

问题描述

给定一个字符串数组 ideas，表示在公司命名过程中使用的名字列表。公司的命名流程如下：

1. 从 ideas 中选择两个不同的名字，称为 ideaA 和 ideaB。
2. 交换 ideaA 和 ideaB 的首字母。
3. 如果得到的两个新名字都不在 ideas 中，那么 ideaA ideaB（串联 ideaA 和 ideaB，中间用一个空格分隔）是一个有效的公司名字。

你的任务是返回不同且有效的公司名字的数量。

示例

示例 1：

输入：

ideas = ["coffee","donuts","time","toffee"]

输出：

6

示例 2：

输入：

ideas = ["lack","back"]

输出：

0

代码实现

class Solution {
public:
    long long distinctNames(vector<string>& ideas) {
        unordered_set<string> groups[26];
        for (auto& s : ideas) {
            groups[s[0] - 'a'].insert(s.substr(1)); // 按照首字母分组
        }

        long long ans = 0;
        for (int a = 0; a < 26; a++) { // 枚举所有组对
            for (int b = 0; b < 26; b++) {
                if (a == b) continue; // 跳过相同组
                int m = 0; // 交集的大小
                for (auto& s : groups[a]) {
                    m += groups[b].count(s); // 计算交集大小
                }
                // 有效的公司名字数量
                ans += (long long)(groups[a].size() - m) * (groups[b].size() - m);
            }
        }
        return ans; 
    }
};

思路解析

1. 字符串分组

首先，我们根据每个名字的首字母将它们分组。使用一个数组 groups，其中 groups[i] 存储以字母 a + i 开头的所有名字的后缀部分。这样，我们能够快速查找哪些名字共享相同的后缀，从而在交换时避免冲突。

2. 枚举组对

接下来，我们需要枚举所有不同首字母的组（即不同的 a 和 b），并计算这两个组之间可以形成的有效公司名字。对于每一对组，我们需要计算它们的交集大小。

3. 计算有效名字数量

对于每一对组 a 和 b，有效名字的计算公式为：

$$\text{有效名字数量}=(\text{组 }a\text{ 的大小}-\text{交集大小})\times (\text{组 }b\text{ 的大小}-\text{交集大小})$$

这个公式考虑了可以与组 a 中的名字交换而不导致重复的情况。

4. 返回结果

最终，我们将所有有效名字的数量累加，返回结果。

结论

通过这种方法，我们可以高效地计算出不同且有效的公司名字的数量，避免了直接暴力枚举的复杂性。这个思路的关键在于利用分组和交集的概念来快速排除无效组合。