java 实现集合分组功能

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'}]}

3.原文连接

https://blog.csdn.net/dalinsi/article/details/78067576

https://blog.csdn.net/u012250875/article/details/55126531

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值