Java中消灭嵌套for循环

心已赠人 2024-03-17 17:38 75阅读 0赞

当两个列表的长度很大时，使用嵌套循环确实可能会导致效率问题。这种情况下，你可以考虑使用Java 8中的流式操作和flatMap方法来优化代码，以实现更高效的笛卡尔积计算。下面是一个优化后的示例代码：

import java.util.ArrayList;
    import java.util.List;
    import java.util.stream.Collectors;
    
    public class CartesianProductExample {
        
        public static class MyObject {
        
            private int value1;
            private int value2;
    
            public MyObject(int value1, int value2) {
        
                this.value1 = value1;
                this.value2 = value2;
            }
    
            public int getValue1() {
        
                return value1;
            }
    
            public void setValue1(int value1) {
        
                this.value1 = value1;
            }
    
            public int getValue2() {
        
                return value2;
            }
    
            public void setValue2(int value2) {
        
                this.value2 = value2;
            }
    
            @Override
            public String toString() {
        
                return "MyObject{" + "value1=" + value1 + ", value2=" + value2 + '}';
            }
        }
    
        public static void main(String[] args) {
        
            List<Integer> list1 = new ArrayList<>();
            List<Integer> list2 = new ArrayList<>();
            // 假设list1包含[1, 2, 3]     
            list1.add(1);
            list1.add(2);
            list1.add(3);
            // 假设list2包含[4, 5]       
            list2.add(4);
            list2.add(5);
            List<MyObject> cartesianProduct = list1.stream().flatMap(value1 -> list2.stream().map(value2 -> new MyObject(value1, value2))).collect(Collectors.toList());
            // 打印结果       
            cartesianProduct.forEach(System.out::println);
        }
    }

在优化后的代码中，我们使用了Java 8提供的流式操作。使用stream()方法将两个列表转换为流，然后使用flatMap()方法将第一个列表中的每个元素与第二个列表中的所有元素进行组合，并映射为一个新的MyObject对象。最后，使用collect()方法将流转换为新的对象列表cartesianProduct。运行优化后的代码将得到与之前相同的输出结果。通过使用流式操作，我们能够以更优雅和高效的方式计算笛卡尔积，并避免了嵌套循环所带来的性能问题。当列表长度较大时，这种优化可以提高代码的执行效率。