题目
给定一个字符串数组ideas表示在公司命名过程中使用的名字列表。公司命名流程如下：
- 从
ideas中选择 2 个不同名字，称为idea_A和idea_B。 - 交换
idea_A和idea_B的首字母。 - 如果得到的两个新名字都不在
ideas中，那么idea_A idea_B（串联idea_A和idea_B，中间用一个空格分隔）是一个有效的公司名字。 - 否则，不是一个有效的名字。
示例 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中的所有字符串互不相同。
解题思路
见代码
代码
cpp
/*
对于每一个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;//将前后的单词交换可以得出一个新的答案
}
};