Java之把一个String数组中有相同共性的元素抽取出来组成新的数组

    /**
     * 方法一
     * 输出map:{101=[101//aaa, 101//ccc], 102=[102//bbb, 102//vvvv], 103=[103//aaa]}
     */
    @Test
    public void testMap1() {
        String[] data = {"101//aaa", "102//bbb", "101//ccc", "103//aaa", "102//vvvv"};
        Map<String, List<String>> mapSave = new HashMap<>();
        for (String str : data) {
            //String key = str.substring(0, str.indexOf("//"));
            String key = StringUtils.substringBefore(str, "//");
            List<String> list = mapSave.get(key);
            //如果map中对应key,创建String数组，并加到map中
            if (list == null || list.isEmpty()) {
                list = new ArrayList<>();
                mapSave.put(key, list);
            }
            //如果map有key对应的string数组，更新数组，为啥map没有put就更新了数组
            list.add(str);
        }
        System.out.println("输出map:" + mapSave);
    }

    /**
     * 方法二
     * 1.原始数组data中的值：101//aaa
     * 2.截取到的字符串：101
     * 4.新创建的数组listStr：[101//aaa]
     * 1.原始数组data中的值：102//bbb
     * 2.截取到的字符串：102
     * 4.新创建的数组listStr：[102//bbb]
     * 1.原始数组data中的值：101//ccc
     * 2.截取到的字符串：101
     * 3.map中有这个key，根据这个key取到的数组：[101//aaa]
     * 4.数组stringList增加一个数据：[101//aaa, 101//ccc]
     * 1.原始数组data中的值：103//aaa
     * 2.截取到的字符串：103
     * 4.新创建的数组listStr：[103//aaa]
     * 1.原始数组data中的值：102//vvvv
     * 2.截取到的字符串：102
     * 3.map中有这个key，根据这个key取到的数组：[102//bbb]
     * 4.数组stringList增加一个数据：[102//bbb, 102//vvvv]
     * 输出map:{101=[101//aaa, 101//ccc], 102=[102//bbb, 102//vvvv], 103=[103//aaa]}
     */
    @Test
    public void testMap2() {
        Map<String, List<String>> mapSave = new HashMap<>();
        String[] data = {"101//aaa", "102//bbb", "101//ccc", "103//aaa", "102//vvvv"};
        for (int i = 0; i < data.length; i++) {
            String a = data[i];
            System.out.println("1.原始数组data中的值：" + a);
            String b = StringUtils.substringBefore(a, "//");
            System.out.println("2.截取到的字符串：" + b);
            //判断map是否有这个key
            boolean flag = mapSave.containsKey(b);
            if (flag) {
                //有这个key
                List<String> stringList = mapSave.get(b);
                System.out.println("3.map中有这个key，根据这个key取到的数组：" + stringList);
                stringList.add(a);
                System.out.println("4.数组stringList增加一个数据：" + stringList);
                mapSave.put(b, stringList);
            } else {
                //没有这个key,创建string数组，并加到map中
                List<String> listStr = new ArrayList<>();
                listStr.add(a);
                System.out.println("4.新创建的数组listStr：" + listStr);
                mapSave.put(b, listStr);
            }
        }
        System.out.println("输出map:" + mapSave);
    }

    /**
     * 方法三
     * 输出map:{101=[101//aaa, 101//ccc], 102=[102//bbb, 102//vvvv], 103=[103//aaa]}
     */
    @Test
    public void testMap3() {
        String[] data = {"101//aaa", "102//bbb", "101//ccc", "103//aaa", "102//vvvv"};
        Map<String, List<String>> mapSave = Arrays.stream(data).collect(Collectors.groupingBy(item -> StringUtils.substringBefore(item, "//")));
        System.out.println("输出map:" + mapSave);
    }