使用Map优化双层For循环

Posted east7

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用Map优化双层For循环相关的知识,希望对你有一定的参考价值。

       笔者在《for循环实战性能优化》中提出了五种提升for循环性能的优化策略,这次我们在其中嵌套循环优化小循环驱动大循环的基础上,借助Map高效的查询性能来优化双层for循环。
      如果小循环和大循环的集合元素数量分别为M和N,则双层For循环的循环次数是M*N,随着M和N的增长,对性能的影响越来越大。因此,本文考虑进一步优化。利用下面的代码来模拟测试两种情况的性能:
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.function.Function;
import java.util.stream.Collectors;

public class ForUpdate {

    public static void main(String[] args) {

//        for (int i = 0; i < 10000; i += 10) {
//            loopGivenNum(i);
//        }
        for (int i = 10000; i < 100000; i += 10000) {
            loopGivenNum(i);
        }
        System.out.println("----- done -----");

    }

    private static void loopGivenNum(int i) {
        List<String> smallLoop = getLoopList(i);
        List<String> bigLoop = getLoopList(2 * i);
        long doByForTimes = doByFor(bigLoop, smallLoop);
        long doByMapTimes = doByMap(bigLoop, smallLoop);
        System.out.println("size " + i + ": " + doByForTimes + "," + doByMapTimes);
    }

    /**
     * 获取循环变量
     * @param size 循环变量元素个数
     */
    private static List<String> getLoopList(int size) {
        List<String> list = new ArrayList<>();
        for (int i = 0; i < size; i++) {
            list.add(String.valueOf(i));
        }
        return list;
    }

    private static long doByFor(List<String> bigLoop, List<String> smallLoop) {
        long startTime = System.currentTimeMillis();
        for (String str1 : smallLoop) {
            for (String str2 : bigLoop) {
                if (str1.equals(str2)) {
                    continue;
                }
            }
        }
        return System.currentTimeMillis() - startTime;
    }

    /**
     * 使用 Map 优化
     * @param bigLoop
     * @param smallLoop
     */
    private static long doByMap(List<String> bigLoop, List<String> smallLoop) {
        long startTime = System.currentTimeMillis();
        // 转换成map
        Map<String, String> loopMap = bigLoop.stream().collect(Collectors.toMap(k -> k, Function.identity()));
        System.out.println(loopMap.size());
        for (String str1 : smallLoop) {
            if (loopMap.containsKey(str1)) {
                continue;
            }
        }
        return System.currentTimeMillis() - startTime;
    }
}

       输出结果:

size 10000: 756,97
size 20000: 3091,8
size 30000: 4342,7
size 40000: 8848,7
size 50000: 16317,7
size 60000: 31652,7
size 70000: 37078,7
     由此可见,数据量越大嵌套For循环执行时间越长,而使用Map后,纵使数据量增长到了20w,执行时间也维持在7ms左右。数据量小的时候,执行结果就不再贴出来了。
     结论:使用Map优化后的方法执行的效率比嵌套循环提高了很多很多。

 

 

 

 

 

以上是关于使用Map优化双层For循环的主要内容,如果未能解决你的问题,请参考以下文章

js优化双层for循环的方法

java双循环list数据保存问题

策略与优化 - 001- 在特定场景下,如何对双层循环进行降级,加速数据匹配?

c语言作业,求大神帮忙:使用双层 for 循环输出九九乘法口诀表

在java中,如何跳出当前的嵌套循环

Linux shell for 循环和双层循环