JAVA 集合 List 分组的两种方法

从网上找了两种方法,效率差不多,这里贴出代码供大家参考

实体类Data

public class Data {

private Long id ;
private Long courseId ;
private String content ;

public Long getId() {
return id;
}

public Data setId(Long id) {
this.id = id;
return this ;
}

public Long getCourseId() {
return courseId;
}

public Data setCourseId(Long courseId) {
this.courseId = courseId;
return this ;
}

public String getContent() {
return content;
}

public Data setContent(String content) {
this.content = content;
return this ;
}

}

排序类

<pre name="code" class="java">import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import com.framework.util.ParamUtils;

public class CommonUtils {

/**
* 分組依據接口，用于集合分組時，獲取分組依據
*
* @author ZhangLiKun
* @title GroupBy
* @date 2013-4-23
*/
public interface GroupBy<T> {
T groupby(Object obj);
}

/**
*
* @param colls
* @param gb
* @return
*/
public static final <T extends Comparable<T>, D> Map<T, List<D>> group(Collection<D> colls, GroupBy<T> gb) {
if (colls == null || colls.isEmpty()) {
System.out.println("分組集合不能為空!");
return null;
}
if (gb == null) {
System.out.println("分組依據接口不能為Null!");
return null;
}
Iterator<D> iter = colls.iterator();
Map<T, List<D>> map = new HashMap<T, List<D>>();
while (iter.hasNext()) {
D d = iter.next();
T t = gb.groupby(d);
if (map.containsKey(t)) {
map.get(t).add(d);
} else {
List<D> list = new ArrayList<D>();
list.add(d);
map.put(t, list);
}
}
return map;
}
/**
* 将List<V>按照V的methodName方法返回值（返回值必须为K类型）分组，合入到Map<K, List<V>>中<br>
* 要保证入参的method必须为V的某一个有返回值的方法，并且该返回值必须为K类型
*
* @param list
*            待分组的列表
* @param map
*            存放分组后的map
* @param clazz
*            泛型V的类型
* @param methodName
*            方法名
*/
public static <K, V> void listGroup2Map(List<V> list, Map<K, List<V>> map, Class<V> clazz, String methodName) {
// 入参非法行校验
if (null == list || null == map || null == clazz || !ParamUtils.chkString(methodName)) {
System.out.print("CommonUtils.listGroup2Map 入参错误，list：" + list + " ；map：" + map + " ；clazz：" + clazz + " ；methodName：" + methodName);
return;
}

// 获取方法
Method method = getMethodByName(clazz, methodName);
// 非空判断
if (null == method) {
return;
}

// 正式分组
listGroup2Map(list, map, method);
}
/**
* 根据类和方法名，获取方法对象
*
* @param clazz
* @param methodName
* @return
*/
public static Method getMethodByName(Class<?> clazz, String methodName) {
Method method = null;
// 入参不能为空
if (null == clazz || !ParamUtils.chkString(methodName)) {
System.out.print("CommonUtils.getMethodByName 入参错误，clazz：" + clazz + " ；methodName：" + methodName);
return method;
}

try {
method = clazz.getDeclaredMethod(methodName);
} catch (Exception e) {
System.out.print("类获取方法失败！");
}

return method;
}
/**
* 将List<V>按照V的某个方法返回值（返回值必须为K类型）分组，合入到Map<K, List<V>>中<br>
* 要保证入参的method必须为V的某一个有返回值的方法，并且该返回值必须为K类型
*
* @param list
*            待分组的列表
* @param map
*            存放分组后的map
* @param method
*            方法
*/
@SuppressWarnings("unchecked")
public static <K, V> void listGroup2Map(List<V> list, Map<K, List<V>> map, Method method) {
// 入参非法行校验
if (null == list || null == map || null == method) {
System.out.print("CommonUtils.listGroup2Map 入参错误，list：" + list + " ；map：" + map + " ；method：" + method);
return;
}

try {
// 开始分组
Object key;
List<V> listTmp;
for (V val : list) {
key = method.invoke(val);
listTmp = map.get(key);
if (null == listTmp) {
listTmp = new ArrayList<V>();
map.put((K) key, listTmp);
}
listTmp.add(val);
}
} catch (Exception e) {
System.out.print("分组失败！");
}
}

}

测试类
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import com.framework.common.CommonUtils.GroupBy;

public class Test {

/**
* @param args
*/
public static void main(String[] args) {
// 准备一个集合

final int loop = 1000 * 1000;
List<Data> list = new ArrayList<Data>(); // size=8 * loop
for (int i = 0; i < loop; i++) {
list.add(new Data().setId(1L).setCourseId(200010L).setContent("AAA"));
list.add(new Data().setId(2L).setCourseId(200010L).setContent("BBB"));
list.add(new Data().setId(3L).setCourseId(200011L).setContent("CCC"));
list.add(new Data().setId(4L).setCourseId(200011L).setContent("DDD"));
list.add(new Data().setId(5L).setCourseId(200010L).setContent("EEE"));
list.add(new Data().setId(6L).setCourseId(200011L).setContent("FFF"));
list.add(new Data().setId(7L).setCourseId(200010L).setContent("GGG"));
list.add(new Data().setId(8L).setCourseId(200012L).setContent("HHH"));
}
// 进行分组 1
long time = System.currentTimeMillis();
Map<Long, List<Data>> map2 = new LinkedHashMap<Long, List<Data>>();
CommonUtils.listGroup2Map(list, map2, Data.class, "getId");// 输入方法名

long duration = System.currentTimeMillis() - time;

System.out.println("分组一执行：" + duration + "毫秒!");

// 分组二
time = System.currentTimeMillis();
Map<Long, List<Data>> map = CommonUtils.group(list, new GroupBy<Long>() {
@Override
public Long groupby(Object obj) {
Data d = (Data) obj;
return d.getCourseId(); // 分组依据为课程ID
}
});

duration = System.currentTimeMillis() - time;

System.out.println("分组二执行：" + duration + "毫秒!");

}

}