【hiho39】二分归并排序之逆序对

最近申请了微软的暑假实习,4号就要在线笔试了,在线测试系统用的是,今天试手做了一道题。

【题目】

原题链接:

输入

第1行:1个整数N,表示数组长度。第2行:N个整数,,表示数组的元素a[i],1≤a[i]≤2^31-1。

输出

第1行:1个整数,表示有多少对元素满足a[i]>a[j],i<j。

样例输入101559614248 709366232 500801802 128741032 1669935692 1993231896 369000208 381379206 962247088 237855491样例输出27

【Java代码】

import java.util.Scanner;import java.math.BigInteger;public class Main {//public int ans = 0;public BigInteger ans = BigInteger.ZERO;public void mergeSort(int[] a, int l, int r) {if (l >= r) return;int mid = (l + r) >> 1;mergeSort(a, l, mid);mergeSort(a, mid + 1, r);merge(a, l, mid, r);}public void merge(int[] a, int l, int mid, int r) {int[] b = new int[r – l + 1];int i = l, j = mid + 1, k = 0;while (i <= mid && j <= r) {if (a[i] <= a[j]) {b[k++] = a[i++];} else {//ans += mid – i + 1;ans = ans.add(new BigInteger(String.valueOf(mid – i + 1)));b[k++] = a[j++];}}while (i <= mid) {b[k++] = a[i++];}while (j <= r) {b[k++] = a[j++];}i = l;j = 0;while (i <= r) {a[i++] = b[j++];}}public static void main(String[] args) {Scanner in = new Scanner(System.in);int N = in.nextInt();int[] a = new int[N];for (int i = 0; i < N; i++) {a[i] = in.nextInt();}Main m = new Main();m.mergeSort(a, 0, N – 1);//System.out.println(m.ans);System.out.println(m.ans.toString());}}

代码中注释部分是第一次提交的代码,得分为80/100,看讨论得知是int会溢出,所以换成BigInteger。

Tips:像这种部分test case通不过时,就要考虑大数溢出的情况。

无神的瞳孔,我迫切想逃离这周遭被钢筋混凝土堆架的城市,

【hiho39】二分归并排序之逆序对

相关文章:

你感兴趣的文章:

标签云: