POJ2299 Ultra-QuickSort (JAVA)

思路是分治，和归并排序一模一样，只是在归并的过程中，顺便统计后半部分序列比前半部分序列小的有多少个

但一直WA，最后是结果数量比较大，会超过int，用long就ac了。。做题真坎坷

贴AC代码

import java.util.*;
public class POJ2299 {

    static int[] arr;

    // 交换数组中的两个元素
    static void swop(int i,int j){
        int temp=arr[i];
        arr[i]=arr[j];
        arr[j]=temp;
    }

    // 一般归并排序写两个函数，我懒得写，感觉合成一个也挺好嘛
    static long compute(int l,int r){
        // 递归结束条件不要忘记了哦
        if(r-l<=1){
            if(arr[r]<arr[l]){
                swop(l,r);
                return 1;
            }else {
                return 0;
            }
        }
        int mid = (l+r)/2;
        // result = 两个数都位于前半部分时，逆序对数量d1 + 两个数都位于后半部分时，逆序对数量d2 +
        // 第一个数字位于前半部分，第二个数字位于后半部分时，逆序对数量count

        // 分治分治，先分后治
        long d1 = compute(l,mid);
        long d2 = compute(mid+1,r);
        // 前半部分长度
        int len1 = mid-l+1;
        // 后半部分长度
        int len2 = r-mid;
        int[] arr1 = new int[len1];
        int[] arr2 = new int[len2];

        // 把两个部分的数组拿出来，等会儿要用
        for(int i=0;i<len1;i++){
            arr1[i] = arr[l+i];
        }

        for(int j=0;j<len2;j++){
            arr2[j] = arr[mid+1+j];
        }

        // 计算第一个数字位于前半部分，第二个数字位于后半部分时，逆序对数量
        int i=0,j=0,k=l;
        // count是结果，必须定义成long，否则WA
        long count=0;
        // 归并过程
        while (i<len1 && j<len2){
            if(arr1[i]<arr2[j]){
                arr[k++]=arr1[i++];
            }else {
                arr[k++]=arr2[j++];
                // 归并过程中，如果前面某个元素大于后面某个元素，说明前面的剩余元素都大于后面的这个元素
                // 所以把前面部分的剩余元素数量给加上
                count+=len1-i;
            }
        }
        if(i>=len1 && j<len2){
            while (j<len2){
                arr[k++]=arr2[j++];
            }
        }
        if(j>=len2 && i<len1){
            while (i<len1){
                arr[k++]=arr1[i++];
            }
        }
        return count+d1+d2;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n=sc.nextInt();
        while (n!=0){
            arr=new int[n];
            for(int i=0;i<n;i++){
                arr[i]=sc.nextInt();
            }
            long result = compute(0,n-1);
            System.out.println(result);

            n=sc.nextInt();
        }
    }
}