leetcode/java/2306-naming-a-company.java at main · neetcode-gh/leetcode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
    public long distinctNames(String[] ideas) {
        Map<Character, Set<String>> map = new HashMap<>();
        for (String s : ideas) {
            char key = s.charAt(0);
            String val = s.substring(1, s.length());
            Set<String> set = map.getOrDefault(key, new HashSet<>());
            set.add(val);
            map.put(key, set);
        }

        long res = 0;
        for (char key1 : map.keySet()) {
            Set<String> set1 = map.get(key1);
            for (char key2 : map.keySet()) {
                if (key1 == key2) continue;
                Set<String> set2 = map.get(key2);
                int overlap = 0;
                for (String s : set1) {
                    if (set2.contains(s)) overlap++;
                }
                res += (set1.size() - overlap) * (set2.size() - overlap);
            }
        }
        return res;
    }
}