70 lines
1.4 KiB
Java
70 lines
1.4 KiB
Java
|
package Maths;
|
||
|
|
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.Arrays;
|
||
|
|
import java.util.Collections;
|
||
|
|
import java.util.HashMap;
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Find the mode of an array of numbers
|
||
|
|
*
|
||
|
|
* The mode of an array of numbers is the most frequently occurring number in the array,
|
||
|
|
* or the most frequently occurring numbers if there are multiple numbers with the same frequency
|
||
|
|
*/
|
||
|
|
public class Mode {
|
||
|
|
|
||
|
|
public static void main(String[] args) {
|
||
|
|
|
||
|
|
/* Test array of integers */
|
||
|
|
assert (mode(new int[]{})) == null;
|
||
|
|
assert Arrays.equals(mode(new int[]{5}), new int[]{5});
|
||
|
|
assert Arrays.equals(mode(new int[]{7, 9, 9, 4, 5, 6, 7, 7, 8}), new int[]{7});
|
||
|
|
assert Arrays.equals(mode(new int[]{7, 9, 9, 4, 5, 6, 7, 7, 9}), new int[]{7, 9});
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
* Find the mode of an array of integers
|
||
|
|
*
|
||
|
|
* @param int[] array of integers
|
||
|
|
* @return int[] mode of the array
|
||
|
|
*/
|
||
|
|
public static int[] mode(int[] numbers) {
|
||
|
|
|
||
|
|
if(numbers.length == 0) return null;
|
||
|
|
|
||
|
|
HashMap<Integer, Integer> count = new HashMap<Integer, Integer>();
|
||
|
|
|
||
|
|
for(int num : numbers) {
|
||
|
|
|
||
|
|
if(count.containsKey(num)) {
|
||
|
|
|
||
|
|
count.put(num, count.get(num) + 1);
|
||
|
|
|
||
|
|
} else {
|
||
|
|
|
||
|
|
count.put(num, 1);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
int max = Collections.max(count.values());
|
||
|
|
ArrayList<Integer> modes = new ArrayList<Integer>();
|
||
|
|
|
||
|
|
for(int num : count.keySet()) {
|
||
|
|
|
||
|
|
if(count.get(num) == max) {
|
||
|
|
|
||
|
|
modes.add(num);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return modes.stream().mapToInt(n -> n).toArray();
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|