C语言优化实例:一种消除嵌套switch-case的巧妙做法

时间:2022-12-13 20:38:58

我们有可能会写出或者遇到类似这样的代码:

 
 
 
C/C++
switch (expr1)
{
case label11:
switch (expr2)
{
case label21:
// do something
break;
case label22:
// do something
break;
default:
// do something
break;
}
break;
case label12:
switch (expr2)
{
case label21:
// do something
break;
case label22:
// do something
break;
default:
// do something
break;
}
break;
case label13:
switch (expr2)
{
case label21:
// do something
break;
case label22:
// do something
break;
default:
// do something
break;
}
break;
default:
// do something
break;
}

这时可以考虑将两个嵌套的switch-case语句通过下面的方式转换为一个switch-case语句:

一般地,expr1和expr2的label情况不会太多,值也不会太大,可以先确定expr2的最大label值,也就能确定要表达expr2所需要的最大位数expr2_max_bit,然后把expr2放到new_expr的低位,把expr1放到new_expr的高位,然后就简化为一层switch-case了。

示例如下:

 
 
 
C/C++
new_expr = expr1 << expr2_max_bit + expr2;
switch (new_expr)
{
case label1:
// do something
break;
case label2:
// do something
break;
case label3:
// do something
break;
case label4:
// do something
break;
case label5:
// do something
break;
case label6:
// do something
break;
default:
// do something
break;
}

注意:通过这种方式并不能消除总的case数,但是可以简化逻辑和代码。