公司命名

/*
题目思路：
1. 使用Java Stream API遍历每一对名字。
2. 交换它们的首字母并检查新名字是否在原始数组中。
3. 如果新名字不在数组中，计数器加一。
4. 返回计数器的值。
*/
import java.util.HashSet;
import java.util.Set;
import java.util.stream.IntStream;

public class Solution {
    public int distinctNames(String[] ideas) {
        Set<String> ideaSet = new HashSet<>();
        for (String idea : ideas) {
            ideaSet.add(idea);
        }

        return (int) IntStream.range(0, ideas.length)
                .boxed()
                .flatMap(i -> IntStream.range(i + 1, ideas.length)
                        .mapToObj(j -> new String[]{ideas[i], ideas[j]}))
                .filter(pair -> {
                    String newIdeaA = pair[1].charAt(0) + pair[0].substring(1);
                    String newIdeaB = pair[0].charAt(0) + pair[1].substring(1);
                    return !ideaSet.contains(newIdeaA) && !ideaSet.contains(newIdeaB);
                })
                .count();
    }
}