Day 0: Mean, Median, and Mode HackerRank Solution


Day 0: Mean, Median, and Mode HackerRank Solution
Source : https://www.hackerrank.com/challenges/s10-basic-statistics



Source : https://www.hackerrank.com/challenges/s10-basic-statistics


Solution


// github.com/RodneyShag
import java.util.Scanner;
import java.util.Arrays;
import java.util.HashMap;
// Time Complexity: O(n log n) due to sorting
public class Solution {
public static void main(String[] args) {
/* Save input */
Scanner scan = new Scanner(System.in);
int size = scan.nextInt();
int [] array = new int[size];
for (int i = 0; i < size; i++) {
array[i] = scan.nextInt();
}
scan.close();
/* Sort array: O(n log n) runtime */
Arrays.sort(array);
/* Calculate Mean */
int total = 0;
for (int num : array) {
total += num;
}
double mean = (double) total / size;
/* Calculate Median */
double median;
if (size % 2 == 0) {
median = (array[size / 2 - 1] + array[size / 2]) / 2.0;
} else {
median = array[size / 2];
}
/* Calculate Mode - if there's a tie, choose the smaller number */
HashMap<Integer, Integer> map = new HashMap();
int maxOccurrences = 0;
int mode = Integer.MAX_VALUE;
for (int num : array) {
map.merge(num, 1, Integer::sum);
int occurrences = map.get(num);
if (occurrences > maxOccurrences || (occurrences == maxOccurrences && num < mode)) {
maxOccurrences = occurrences;
mode = num;
}
}
/* Print results */
System.out.println(mean);
System.out.println(median);
System.out.println(mode);
}
}

No comments:

Post a Comment