如何过滤数组中的正方形[关闭]

Question

我是 Java 编码新手，需要一些帮助。

我需要一种方法来过滤数组中的平方数，例如（4、9、16、25），然后将它们保存在新数组中而不替换旧数组。

不幸的是，我没有办法 :/ 。

这是我自己的一些想法：

public static int[] arrayFilter(int originalArray[]) {
    int[] filterArray = generateCopy(originalArray);
    for(int index = 0; index < filterArray.length; index++) {

        }
    }

}

感谢您的帮助:)

Answer 1

那么，你遍历数组并检查平方根的平方是否与原始数字相同。如果是这样，则将其添加到列表中，完成后将其转换为数组。

int[] numbers = new int[] { 1,2,3,4,5,6,7,8,9,10 };
List<Integer> resultList = new ArrayList<>();
for (int number: numbers) {
    int sqrRoot = (int) Math.round(Math.sqrt(number));
    if (sqrRoot * sqrRoot == number) {
        resultList.add(number);
    }
}

或者如果您不能使用List（请参阅下面原始问题作者的评论）...

int[] numbers = new int[] { 1,2,3,4,5,6,7,8,9,10 };
int[] initResult = new int[numbers.length];
int numSquares = 0;
for (int i = 0; i < numbers.length; i++) {
    int sqrRoot = (int) Math.round(Math.sqrt(numbers[i]));
    if (sqrRoot * sqrRoot == numbers[i]) {
        initResult[numSquares] = numbers[i];
        numSquares++;
    }
}
int[] result = new int[numSquares];
for (int i = 0; i < numSquares; i++) {
    result[i] = initResult[i];
}

我计算平方根如下(int) Math.round(Math.sqrt(numbers[i])) 的原因是为了防止floating point arithmetic 问题。这样，整数 4 的平方根将始终是值为 2 的整数。

Answer 2

这里是一个使用带有过滤器的 Streams 来创建新输出数组的示例。

public class Squares {
    public static void main(String... args) {
        double[] numbers = {1, 2, 3, 4, 4.4, 100, 2500, 2500.1};
        double[] squares = Arrays.stream(numbers)
                .filter(Squares::isSquare)
                .toArray();
        System.out.println(Arrays.toString(squares));
    }

    private static boolean isSquare(double d) {
        double sr = Math.sqrt(d);
        return sr - Math.floor(sr) == 0;
    }
}