java 实现集合分组功能

最新推荐文章于 2024-08-21 02:26:10 发布

再学HelloWorld

最新推荐文章于 2024-08-21 02:26:10 发布

阅读量1w

点赞数 1

分类专栏： java

java 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

1.代码

package com.xtm.java.test;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Comparator;
import java.util.List;

/**
 * 分组实现
 */
public class Test{
    //实体类
    class Apple {
        String color;
        int weight;

        Apple(String color, int weight) {
            super();
            this.color = color;
            this.weight = weight;
        }

        @Override
        public String toString() {
            return "Apple{" +
                    "颜'" + color + '\'' +
                    ", 重：" + weight +
                    '}';
        }
    }


    public static void main(String args[]){
        List<Apple> list = new ArrayList<>();
        list.add(new Test().new Apple("红", 205));
        list.add(new Test().new Apple("绿", 248));
        list.add(new Test().new Apple("红", 131));
        list.add(new Test().new Apple("黄", 119));
        list.add(new Test().new Apple("绿", 153));
        list.add(new Test().new Apple("黄", 224));
        //按颜色分组
        List<List<Apple>> byColorList = byColorGroupApple(list, Comparator.comparing(t -> t.color));
        printList(byColorList);
        //按重量级分组
        List<List<Apple>> byWeightList = byColorGroupApple(list, (t1, t2) -> t1.weight/100==t2.weight/100?0:1);
        printList(byWeightList);
    }

    private static void printList(List<List<Apple>> byColorList) {
        System.out.println("分组后："+byColorList);
    }

    /**
     * 根据传入的比较器进行集合的分组
     * @param data 需要分组的元素集合
     * @param c 传入的比较器
     * @param <T> 元素
     * @return 返回分组后的集合列表
     */
    private static <T> List<List<T>> byColorGroupApple(Collection<T> data, Comparator<? super T> c) {//方法上使用泛型 记得在返回值前加<T>
        List<List<T>> result = new ArrayList<>();
        for (T t : data) {//1.循环取出集合中的每个元素
            boolean isSameGroup = false;//2.标志为不是同组
            for (List<T> aResult : result) {//4.循环查找当前元素是否属于某个已创建的组
                if (c.compare(t, aResult.get(0)) == 0) {//aResult.get(0)表示：只要当前元素和某个组的第一个元素通过比较器比较相等则属于该组
                    isSameGroup = true;//5.查询到当前元素属于某个组则设置标志位，不让其创键新组
                    aResult.add(t);//6.把当前元素添加到当前组
                    break;
                }
            }
            if (!isSameGroup) {//3.不属于任何组的则创建一个新组，并把元素添加到该组
                // 创建
                List<T> innerList = new ArrayList();
                innerList.add(t);
                result.add(innerList);
            }
        }
        return result;
    }

}

代码2：java8

package com.xtm.java.test;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import static java.util.stream.Collectors.groupingBy;

public class Student {

    private int id;
    private String sex;
    private String name;

    public static void main(String[] args) {
        List<Student> list = new ArrayList();
        list.add(new Student(1,"男","zhangsan"));
        list.add(new Student(2,"女","lisi"));
        list.add(new Student(3,"男","wangsu"));
        list.add(new Student(2,"女","zhaoliu"));
        list.add(new Student(1,"男","zhuoqi"));
        list.add(new Student(3,"女","cangkong"));

        //将所有的学生按照性别进行分组
        Map<String, List<Student>> collect = list.stream().collect(Collectors.groupingBy(Student::getSex));

        //用方法引用可读性更好
        Map<String, List<Student>> collect2 = list.stream().collect(groupingBy(Student::getSex));

        //按照Id分组
        Map<Integer, List<Student>> collect3 = list.stream().collect(groupingBy(Student::getId));

        //自定义函数分组
        Map<Integer, List<Student>> collect4 = list.stream().collect(groupingBy(stu -> stu.hashCode() % 5));

        System.out.println(collect);
        System.out.println(collect2);
        System.out.println(collect3);
        System.out.println(collect4);
    }


    public Student(int id, String sex, String name) {
        this.id = id;
        this.sex = sex;
        this.name = name;
    }

    public int getId() {
        return id;
    }

    public String getSex() {
        return sex;
    }

    public String getName() {
        return name;
    }

    @Override
    public String toString() {
        return "Student{" +
                "id=" + id +
                ", sex='" + sex + '\'' +
                ", name='" + name + '\'' +
                '}';
    }
}

2.运行结果

分组后：[[Apple{颜'红', 重：205}, Apple{颜'红', 重：131}], [Apple{颜'绿', 重：248}, Apple{颜'绿', 重：153}], [Apple{颜'黄', 重：119}, Apple{颜'黄', 重：224}]]
分组后：[[Apple{颜'红', 重：205}, Apple{颜'绿', 重：248}, Apple{颜'黄', 重：224}], [Apple{颜'红', 重：131}, Apple{颜'黄', 重：119}, Apple{颜'绿', 重：153}]]

{女=[Student{id=2, sex='女', name='lisi'}, Student{id=2, sex='女', name='zhaoliu'}], 男=[Student{id=1, sex='男', name='zhangsan'}, Student{id=3, sex='男', name='wangsu'}, Student{id=1, sex='男', name='zhuoqi'}, Student{id=3, sex='男', name='cangkong'}]}
{女=[Student{id=2, sex='女', name='lisi'}, Student{id=2, sex='女', name='zhaoliu'}], 男=[Student{id=1, sex='男', name='zhangsan'}, Student{id=3, sex='男', name='wangsu'}, Student{id=1, sex='男', name='zhuoqi'}, Student{id=3, sex='男', name='cangkong'}]}
{1=[Student{id=1, sex='男', name='zhangsan'}, Student{id=1, sex='男', name='zhuoqi'}], 2=[Student{id=2, sex='女', name='lisi'}, Student{id=2, sex='女', name='zhaoliu'}], 3=[Student{id=3, sex='男', name='wangsu'}, Student{id=3, sex='男', name='cangkong'}]}
{0=[Student{id=2, sex='女', name='lisi'}, Student{id=2, sex='女', name='zhaoliu'}], 1=[Student{id=3, sex='男', name='wangsu'}], 2=[Student{id=1, sex='男', name='zhangsan'}, Student{id=1, sex='男', name='zhuoqi'}], 4=[Student{id=3, sex='男', name='cangkong'}]}