将一个List<T>集合拆分成List<List<T>>集合

/**
 * 将一组数据固定分组,每组n个元素
 *
 * @param <T> 泛型对象
 * @param resList 需要拆分的集合
 * @param subListLength 每个子集合的元素个数
 * @return 返回拆分后的各个集合组成的列表
 * 代码里面用到了guava和common的结合工具类
 **/
public static <T> List<List<T>> split(List<T> resList, int subListLength) {
   if (CollectionUtils.isEmpty(resList) || subListLength <= 0) {
      return Lists.newArrayList();
   }
   List<List<T>> ret = Lists.newArrayList();
   int size = resList.size();
   if (size <= subListLength) {
      // 数据量不足 subListLength 指定的大小
      ret.add(resList);
   } else {
      int pre = size / subListLength;
      int last = size % subListLength;
      // 前面pre个集合,每个大小都是 subListLength 个元素
      for (int i = 0; i < pre; i++) {
         List<T> itemList = Lists.newArrayList();
         for (int j = 0; j < subListLength; j++) {
            itemList.add(resList.get(i * subListLength + j));
         }
         ret.add(itemList);
      }
      // last的进行处理
      if (last > 0) {
         List<T> itemList = Lists.newArrayList();
         for (int i = 0; i < last; i++) {
            itemList.add(resList.get(pre * subListLength + i));
         }
         ret.add(itemList);
      }
   }
   return ret;
}

/**
 * 将一组数据平均分成n组
 *
 * @param source 要分组的数据源
 * @param n      平均分成n组
 * @param <T>
 * @return
 */
public static <T> List<List<T>> averageAssign(List<T> source, int n) {
   List<List<T>> result = new ArrayList<List<T>>();
   int remainder = source.size() % n;  //(先计算出余数)
   int number = source.size() / n;  //然后是商
   int offset = 0;//偏移量
   for (int i = 0; i < n; i++) {
      List<T> value = null;
      if (remainder > 0) {
         value = source.subList(i * number + offset, (i + 1) * number + offset + 1);
         remainder--;
         offset++;
      } else {
         value = source.subList(i * number + offset, (i + 1) * number + offset);
      }
      result.add(value);
   }
   return result;
}

// 运行代码
public static void main(String[] args) {
   List<String> list = new ArrayList<>();
   list.add("1");
   list.add("2");
   list.add("3");
   list.add("4");
   list.add("5");
   list.add("6");
   list.add("7");
   //将一组数据平均分成n组
   List<List<String>> temps = averageAssign(list, 4);
   int j = 0;
   for (List<String> obj : temps) {
      System.out.println(String.format("row:%s -> size:%s,data:%s", ++j, obj.size(), obj));
   }

   //将一组数据固定分组,每组n个元素
   List<List<String>> temps2 = split(list, 4);
   int k = 0;
   System.err.println("====================================================================");
   System.err.println("====================================================================");
   System.err.println("====================================================================");
   for (List<String> obj : temps2) {
      System.out.println(String.format("row:%s -> size:%s,data:%s", ++k, obj.size(), obj));
   }
}

猜你喜欢

转载自blog.csdn.net/yyongsheng/article/details/129043618