• Split / Partition a collection into smaller collections


     

    Partition a collection into smaller collections

    This article describes how to partition a collections into a given number of smaller collections.

     

    1. Partition a collection

    Sometime you want split a collection into smaller collections. The following gives an example how to do this.

    Tip

     Please note that this example is an adjusted example of the Google collection and more or less the same as Code Ranch discussion The Google source code can be found here Original source code

    The test code also demonstrate how to calculate the number of elements which should go into one bucket in case you want to have a fixed number of buckets.

     

    2. Partition collection in Java

    2.1. Implementation

    Create a Java project "de.vogella.algorithms.partitioncollection".

    Create the following program.

    package de.vogella.algorithms.partitioncollection;
    
    import java.util.AbstractList;
    import java.util.List;
    
    
    public class MyPartition {
    /** * Returns consecutive {@linkplain List#subList(int, int) sublists} of a list, * each of the same size (the final list may be smaller). For example, * partitioning a list containing {@code [a, b, c, d, e]} with a partition * size of 3 yields {@code [[a, b, c], [d, e]]} -- an outer list containing * two inner lists of three and two elements, all in the original order. * * <p>The outer list is unmodifiable, but reflects the latest state of the * source list. The inner lists are sublist views of the original list, * produced on demand using {@link List#subList(int, int)}, and are subject * to all the usual caveats about modification as explained in that API. * * * Adapted from http://code.google.com/p/google-collections/ * * @param list the list to return consecutive sublists of * @param size the desired size of each sublist (the last may be * smaller) * @return a list of consecutive sublists * @throws IllegalArgumentException if {@code partitionSize} is nonpositive * */
    
    
        
        public static <T> List<List<T>> partition(List<T> list, int size) {
       
         if (list == null)
            throw new NullPointerException("'list' must not be null");
          if (!(size > 0))
            throw new IllegalArgumentException("'size' must be greater than 0");
    
          return new Partition<T>(list, size);
        }
    
        private static class Partition<T> extends AbstractList<List<T>> {
    
          final List<T> list;
          final int size;
    
          Partition(List<T> list, int size) {
            this.list = list;
            this.size = size;
          }
    
          @Override
          public List<T> get(int index) {
            int listSize = size();
            if (listSize < 0)
              throw new IllegalArgumentException("negative size: " + listSize);
            if (index < 0)
              throw new IndexOutOfBoundsException("index " + index + " must not be negative");
            if (index >= listSize)
              throw new IndexOutOfBoundsException("index " + index + " must be less than size " + listSize);
            int start = index * size;
            int end = Math.min(start + size, list.size());
            return list.subList(start, end);
          }
    
          @Override
          public int size() {
            return (list.size() + size - 1) / size;
          }
    
          @Override
          public boolean isEmpty() {
            return list.isEmpty();
          }
        }
    
    
    } 
     
    

      

    2.2. Test

    You can use the following JUnit test to validate the result.

    package de.vogella.algorithms.partitioncollection;
    
    import java.util.ArrayList;
    import java.util.List;
    
    import org.junit.Test;
    
    import static org.junit.Assert.assertTrue;
    
    
    public class MyPartitionTest {
      @Test
      public void partitiontest1() {
        List<String> list = new ArrayList<String>();
        list.add("one");
        list.add("two");
        list.add("three");
        list.add("four");
        list.add("five");
        list.add("six");
        list.add("seven");
        list.add("eight");
        list.add("nine");
        list.add("ten");
        list.add("eleven");
        List<List<String>> partition = MyPartition.partition(list, 1);
        System.out.println(partition.get(2).size()); 
        assertTrue(partition.size()==11);
        assertTrue(partition.get(0).size()==1);
        partition = MyPartition.partition(list, 7);
        assertTrue(partition.size()==2);
        assertTrue(partition.get(0).size()==7);
        assertTrue(partition.get(1).size()==4);
        
        // now let assume you want to have x number of buckets
        // How many elements must placed in a list?
        // Take x as 3
        
        int buckets = 3;
        int divide = list.size() / buckets;  
        if (list.size() % buckets !=0){
          divide ++;
        }
        System.out.println("Max. number of element in each bucket " + divide);
        partition = MyPartition.partition(list, divide);
        assertTrue(partition.size()==buckets);
      }
    } 
     
    

      

  • 相关阅读:
    C语言预处理
    C语言结构体对齐
    C语言共用体、大小端、枚举
    C语言内存分配方法。
    C与指针(结构体指针,函数指针,数组指针,指针数组)定义与使用
    C语言中函数的传入值与传出值
    #define与typedef在重定义类型中的区别
    宏定义在位运算中的运用
    wait函数
    exit()与_exit()区别
  • 原文地址:https://www.cnblogs.com/hephec/p/4579648.html
Copyright © 2020-2023  润新知