java:10 是一个非常特殊的数,它可以表示成两个非负整数的平方和 9 = 3 * 3 + 0 * 0。  

10 是一个非常特殊的数,它可以表示成两个非负整数的平方和,10 = 3 * 3 + 1 * 1。
9 也是同样特殊的数,它可以表示成 9 = 3 * 3 + 0 * 0。
  请问,在 1 到 2021 中有多少个这样的数?
  请注意,有的数有多种表示方法,例如 25 = 5 * 5 + 0 * 0 = 3 * 3 + 4 * 4,在算答案时只算一次。
java可以用3重for循环暴力出所有这样的数,但是会有重复的数字,用java解决,有更好的办法吗

img


三重for为什么会有重复的

    int res = 0;
    int sqrt = (int)Math.sqrt(2021);
    for (int i=1;i<=2021;i++){
        for (int j=0;j<=sqrt;j++){
            int temp = i-j*j;       //  看得到的数开平方取整后再平方再加j平方能否得到i
            int tempsqrt = (int)Math.sqrt(temp);
            if (tempsqrt*tempsqrt+j*j==i){
                res++;
                break;
            }
        }
    }
    System.out.println(res);