2306. 公司命名（24.9.25）

题目

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

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

示例 1：

• 输入：ideas=["coffee","donuts","time","toffee"]
• 输出：6
  • 解释：下面列出一些有效的选择方案：
    • (“coffee”, “donuts”): 对应的公司名字是 “doffee conuts”。
    • (“donuts”, “coffee”): 对应的公司名字是 “conuts doffee”。
    • (“donuts”, “time”): 对应的公司名字是 “tonuts dime”。
    • (“donuts”, “toffee”): 对应的公司名字是 “tonuts doffee”。
    • (“time”, “donuts”): 对应的公司名字是 “dime tonuts”。
    • (“toffee”, “donuts”): 对应的公司名字是 “doffee tonuts”。
  • 因此：总共有 6 个不同的公司名字。
  • 下面列出一些无效的选择方案：
    • (“coffee”, “time”):在原数组中存在交换后形成的名字"toffee"。
    • (“time”, “toffee”): 在原数组中存在交换后形成的两个名字。
    • (“coffee”, “toffee”): 在原数组中存在交换后形成的两个名字。

示例 2：

• 输入：ideas = ["lack","back"]
• 输出：0
  • 解释：不存在有效的选择方案。因此，返回 0。

提示信息：

• 条件：2 <= ideas.length <= 5 * 10^4，即数组ideas的长度在 2 到 50000 之间。
• 条件：1 <= ideas[i].length <= 10，即数组中每个字符串的长度在 1 到 10 之间。
• 条件：ideas[i]由小写英文字母组成。
• 条件：ideas中的所有字符串互不相同。

解题思路

见代码

代码

/*
对于每一个idea而言：我们可以分成两个部分，第一部分是首字母的部分，第二个部分是除第一个字母意外大部分
*/
class Solution {
public:
    long long distinctNames(vector<string>& ideas) {
        long long ans=0;
        unordered_set<string> h[26];//用于记录第一部分和第二部分
        for(string& s:ideas){
            h[s[0]-'a'].insert(s.substr(1));
        }
        //对每一种组合进行枚举
        for(int f=1;f<26;f++)
        {

            for(int s=0;s<f;s++)
            {
                int sum=0;//用于统计交集的部分

                for(auto& str:h[f])
                {
                    sum+=h[s].count(str);
                }

                ans+=(long long) (h[f].size()-sum)*(h[s].size()-sum);//除去交际部分可以得出，不重合部分的排列组合
            }

        }
        
        return ans*2;//将前后的单词交换可以得出一个新的答案
       
    }
};