正方形数组的数目

/*
题目思路：
1. 使用递归生成所有排列。
2. 使用流处理检查每个排列的相邻元素之和是否为完全平方数。
3. 使用集合来存储唯一的排列结果。
*/

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class SquareArrayStream {
    public int numSquarefulPerms(int[] A) {
        Set<List<Integer>> uniquePerms = new HashSet<>();
        permute(A, 0, uniquePerms);
        return (int) uniquePerms.stream()
                .filter(this::isSquareful)
                .count();
    }

    private void permute(int[] A, int start, Set<List<Integer>> result) {
        if (start == A.length) {
            result.add(Arrays.stream(A).boxed().collect(Collectors.toList()));
        } else {
            IntStream.range(start, A.length).forEach(i -> {
                swap(A, i, start);
                permute(A, start + 1, result);
                swap(A, i, start);
            });
        }
    }

    private void swap(int[] A, int i, int j) {
        int temp = A[i];
        A[i] = A[j];
        A[j] = temp;
    }

    private boolean isSquareful(List<Integer> list) {
        return IntStream.range(0, list.size() - 1)
                .allMatch(i -> {
                    int sum = list.get(i) + list.get(i + 1);
                    int sqrt = (int) Math.sqrt(sum);
                    return sqrt * sqrt == sum;
                });
    }
}