两数的平方和

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("]");
    }
}