围绕密钥拆分排序数组的最有效方法

时间:2017-04-14 03:41:54

标签: java arrays algorithm split

考虑我有一个从1到10的排序数字数组。 我想基于关键元素拆分数组。 比如说:

   Key  Transformed Array
    1   [2-10]
    5   [2-4],[6-10]
    6   [2-4],[7-10]
    7   [2-4],[8-10]
    8   [2-4],[9-10]
    10  [2-4],[9-9]

如果键是连续的,我不打算迭代地做, 意思,

Keys 5,6,7 and 8 have to be processed in one step breaking 
[2-4],[6-10] into [2-4],[9-10]

1 个答案:

答案 0 :(得分:0)

最简单的选择是使用Collectors.partitioningBy方法以及Java 8的StreamsCollectors

raw_input

在旁注中,我个人只会使用Python 3.x代替import java.util.stream.Collectors; import java.util.Map; import java.util.List; import java.util.Arrays; public class Partition { public static String join(int[] a, String d) { return Arrays.stream(a).mapToObj(String::valueOf) .collect(Collectors.joining(d)); } public static void main(String[] args) { int[] v = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10}; int k = 4; Map<Boolean, List<Integer>> partitions = Arrays.stream(v) .boxed() .filter(n -> n != k) .collect(Collectors.partitioningBy(n -> n < k)); int[] vlt = partitions.get(true).stream() .mapToInt(Integer::intValue).toArray(); int[] vgt = partitions.get(false).stream() .mapToInt(Integer::intValue).toArray(); System.out.println("vlt: {" + join(vlt, ", ") + "}"); System.out.println("vgt: {" + join(vgt, ", ") + "}"); } } 并跳过转换,如果这是一个选项。