hackerrank 的 MaxDiff 问题 - 优化代码作为它的超时

Question

下面是我在hackerrank 上对maxdiff 问题的实现。 它通过了所有测试用例，除了 1 给出超时。 帮我优化一下。

import java.io.*;
import java.util.*;

public class Solution {

    private static final Scanner scanner = new Scanner(System.in);
    public static void main(String[] args) {
        /* Enter your code here. Read input from STDIN. Print output to STDOUT. Your class should be named Solution. */
        int n = Integer.parseInt(scanner.nextLine());
        if(n==0 || n==1)
            System.out.println(0);;
        String s=scanner.nextLine();
        String ar[] = s.split("\\s+");
        int[] a=new int[n];
        for(int i=0;i<n;i++)
        {
            a[i]=Integer.parseInt(ar[i]);
        }
        int max, min;
        long sum=0;
        for(int i=0;i<n;i++)
        {
            max=a[i];
            min=a[i];

            for(int j=i+1;j<n;j++)
            {
                if(a[j]>max)
                    max=a[j];
                if(a[j]<min)
                    min=a[j];

                sum+=max-min;
            }
        }

        System.out.println(sum);
    }
}

失败的测试用例太大，无法在此处附加。

Answer 1

我看到一个问题。

问题在于处理 n == 0 和 n == 1。您应该在打印 0 后返回：这是您应该拥有的。

if (n==0 || n==1) {
    if (n==1)
        scanner.nextLine();
    System.out.println(0);
    return;
}