C/C++中的switch语句优化效率的两种方式

简介

许多编程语言，如C/C++、C#、Java和Pascal，都提供了switch语句，让我们实现选择逻辑。在某些情况下，它是if-then-else的一个很好的替代品，使代码更清晰、更易读。在实践中使用switch时，你可能想知道。

• switch块在运行时是如何执行的？

• 对于一长串的条件，它的运行速度是否比if-then-else快？

• 对于n个条件，switch的时间复杂度是多少？

C/C++标准定义了语言元素的规范，但它并没有说如何实现switch语句。只要符合标准，每个厂商都可以自由地使用任何实现。本文讨论了在gcc中运行switch语句时会发生什么，并列举了几个不同条件下的例子。通过godbolt来呈现汇编后的逻辑。

The Jump Table

第一个例子是一个简单的switch语句，case的范围较为集中。首先看下汇编代码：

这里用的是gcc 11.2没有开编译优化，可以看到，汇编代码需要做两层映射，才能达到需求。第一层是汇编的42行，根据rax（可以理解为变量i）作为偏移量，跳转到.L7对应的位置。第二层是根据.L7中对应的映射，跳转到具体的执行case后面逻辑的代码，如.L15为先调用f1()，再break，对应源码的第10行。

上述例子中，switch语句的时间复杂度为O(1)，因为可以直接通过jump table跳转到对应的case。

在这里可以发现，本质上使用jump table来进行跳转是一种用空间换时间的方法，因为映射表的大小会和case的范围正相关，考虑一种极端的例子，如果case只有三种情况：0、10000、default，这时候如果再用jump table，会消耗大量的text空间，还不如直接用逐个比较的方法。以下的例子就说明了，在case的范围较大时，gcc默认使用逐个比较的方法来编译switch-case语句。

这例子中，switch语句的时间复杂度就退化到了为O(n)，实质就是在做遍历case的操作。

Using Binary Search

switch语句和if语句一个不同的点在于，switch语句只能对一个变量进行范围上的划分，而if语句内的判断条件可以表达更丰富的逻辑。switch的这个特点也带来了一个优化的空间，那就是使用二分查找的方式去定位到具体的case。还是上面的例子，如果开启了编译优化选项-O，编译器将使用二分查找的方式来优化switch语句。

先是已700为基准，等于700的直接跳转到.L8执行f2()，大于700的跳转到.L9，再去与800、900进行对比。小于700的与250、500进行对比。小于250的再与100、200进行对比。上述流程是一个典型的二分查找过程。因此将时间复杂度从O(n)优化到了O(logn)，当case项很多并且值比较分散时，这是一个相当可观的优化。

总结

switch语句是一个典型的分支查找语句，由于其结构特点，它的逻辑可读性强，便于理解。由于其特点，编译器可以对其进行特定的优化。在case范围较为集中时，编译器会使用jump table的方式，通过相对偏移量，多添加一层映射，使其能在O(1)的时间复杂度内到达匹配的case所对应的代码块。在case范围较为分散的情况下，编译器提供了二分查找的方式（需开启编译优化）来搜索对应的case，O(logn)的时间复杂度相较原始O(n)的方法也有较大的提升。

如果大家对文章内容有疑问或建议，欢迎评论区留言，我们可以一起讨论，共同提高
转载请标明原作者:https://blog.csdn.net/qq_36622751