力扣每日一题 公司命名 集合 找规律

Problem: 2306. 公司命名

👨‍🏫 灵神题解

class Solution {
    public long distinctNames(String[] ideas) {
        // 创建一个大小为26的HashSet数组，用于存储每个首字母对应的字符串集合
        Set<String>[] groups = new HashSet[26];
        Arrays.setAll(groups, i -> new HashSet<>());
        long ans = 0; // 初始化答案计数器

        // 遍历字符串数组，将每个字符串按首字母分类存储到对应的HashSet中
        for (String s : ideas) {
            groups[s.charAt(0) - 'a'].add(s.substring(1));
        }

        // 遍历所有可能的首字母组合（a和b），a > b
        for (int a = 1; a < 26; a++) {
            for (int b = 0; b < a; b++) {
                int m = 0; // 初始化交集计数器
                // 计算两个首字母集合的交集大小
                for (String s : groups[a]) {
                    if (groups[b].contains(s))
                        m++;
                }
                // 计算可以交换首字母的字符串对数量
                // (a集合中非交集部分的大小) * (b集合中非交集部分的大小)
                ans += (long) (groups[a].size() - m) * (groups[b].size() - m);
            }
        }

        // 由于每对字符串可以以两种方式交换（ab和ba），所以最终答案乘以2
        return ans * 2;
    }
}

原文地址：https://blog.csdn.net/lt6666678/article/details/142536073

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！