-
Notifications
You must be signed in to change notification settings - Fork 79
/
MergeSort.js
86 lines (60 loc) · 2.01 KB
/
MergeSort.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
'use strict';
const fs = require('fs');
process.stdin.resume();
process.stdin.setEncoding('utf-8');
let inputString = '';
let currentLine = 0;
process.stdin.on('data', inputStdin => {
inputString += inputStdin;
});
process.stdin.on('end', function() {
inputString = inputString.replace(/\s*$/, '')
.split('\n')
.map(str => str.replace(/\s*$/, ''));
main();
});
function readLine() {
return inputString[currentLine++];
}
// Complete the countInversions function below.
function mergeSort(arr) {
if(arr.length == 1)
return [arr, 0];
let half = Math.floor(arr.length / 2);
let [leftHalf, leftswap] = mergeSort(arr.slice(0,half));
let [rightHalf, rightswap] = mergeSort(arr.slice(half));
let leftPointer = 0;
let rightPointer = 0;
let swap = leftswap + rightswap;
let sorted = [];
while(leftPointer < leftHalf.length || rightPointer < rightHalf.length){
if(leftPointer != leftHalf.length && rightPointer != rightHalf.length){
if(leftHalf[leftPointer] <= rightHalf[rightPointer]){
sorted.push(leftHalf[leftPointer++]);
}else if(leftHalf[leftPointer] > rightHalf[rightPointer]){
swap += leftHalf.length - leftPointer;
sorted.push(rightHalf[rightPointer++]);
}
}else if (leftPointer == leftHalf.length){
sorted.push(rightHalf[rightPointer++]);
}else{
sorted.push(leftHalf[leftPointer++]);
}
}
return [sorted, swap];
}
function countInversions(arr) {
let [_, totalInversions] = mergeSort(arr);
return totalInversions;
}
function main() {
const ws = fs.createWriteStream(process.env.OUTPUT_PATH);
const t = parseInt(readLine(), 10);
for (let tItr = 0; tItr < t; tItr++) {
const n = parseInt(readLine(), 10);
const arr = readLine().split(' ').map(arrTemp => parseInt(arrTemp, 10));
const result = countInversions(arr);
ws.write(result + '\n');
}
ws.end();
}