HackerEarth 练习时间不多了

Question

我从 HackerEarth解决了以下问题。所有测试用例都是正确的，除了最后一个因为它超时了。我尝试优化我的解决方案，但我无法更好地优化它。这是我的解决方案：import java.util.HashSet;import java.util.Scanner;import java.util.Set;class TestClass {    public static void main(String args[]) throws Exception {        Scanner sc = new Scanner(System.in);        Set<Integer> perfectNumbers = new HashSet<>();        for (int i = 0; i <= 20; i++) {            perfectNumbers.add(i * i * i);        }        for (int i = 1; i <= 44; i++) {            perfectNumbers.add(i * i);        }        int t = sc.nextInt();        for (int k = 0; k < t; k++) {            int db = 0;            int n = sc.nextInt();            int[] a = new int[1001];            int[] b = new int[1001];            int[] numbers = new int[n];            for (int j = 0; j < n; j++) {                int x = sc.nextInt();                numbers[j] = x;                for (Integer perfect : perfectNumbers) {                    if (x == perfect - x) {                        b[x]++;                    } else if (perfect - x >= 0 && perfect - x <= 1000)                        a[perfect - x]++;                }            }            for (int j = 0; j < n; j++) {                db += a[numbers[j]];            }            for (int j = 0; j <= 1000; j++) {                if (b[j] > 1) {                    db += b[j] * (b[j] - 1);                }            }            System.out.println(db / 2);                   }    }}