两数的平方和

Posted 2021-01-24 earthhouge

题目描述：判断一个给定的数是否为两个数的平方和。

题目分析：使用双指针，一个数从1开始，一个数从给定数的算术平方根开始。较小的数每次累加1，较大的数每次减1。

• 如果两个数的平方和 sum == target，那么得到要求的结果；
• 如果 sum > target，移动较大的元素，使 sum 变小一些；
• 如果 sum < target，移动较小的元素，使 sum 变大一些。

代码如下：

public class DoublePointer {
    public static void main(String[] args){
        print(judgeSquareSum(5));
    }

    public static int[] judgeSquareSum(int c) {
        int i = 0, j = (int) Math.sqrt(c);
        while (i <= j) {
            int powSum = i * i + j * j;
            if (powSum == c) {
                return new int[]{i, j};
            } else if (powSum > c) {
                j--;
            } else {
                i++;
            }
        }
        return null;
    }

    public static void print(int[] arr){
        System.out.print("[");
        for (int i = 0; i < arr.length; i++){
            if (i!=arr.length-1){
                System.out.print(arr[i] + ",");
            }else{
                System.out.print(arr[i]);
            }

        }
        System.out.print("]");
    }
}

执行结果：[1,2]