数学运算比IF要快

问题

虽然很早就知道，CPU在处理 if 这样的判断语句时，使用了预测的技术，所以如果条件总是一个结果，效率就很好。反过来说，如果你使用数学运算避免 if 判断，那么就意味着性能一定比 if 要好。

方案1

今天正好有个函数遇到这个问题，所以我就正好测试以下。

待测试的方法是获取一个int32的数据中，有多少个位是 1，我的方案是将 一个int32拆开成 4个字节，然后一一判断。下面是使用 if 判断的方案 （那个  ? : 三元运算符就是 if 语句）。

static int getInt32TrueCount(int value)
{
    if (value == 0)
    {
        return 0;
    }

    return getByteTrueCount(value & 0xFF) +
        getByteTrueCount((value >> 8) & 0xFF) +
        getByteTrueCount((value >> 16) & 0xFF) +
        getByteTrueCount((value >> 24) & 0xFF);
}

static int getByteTrueCount(int value)
{
    if (value == 0)
    {
        return 0;
    }

    int a = (value & 0x1) == 0 ? 0 : 1;
    int b = (value & 0x2) == 0 ? 0 : 1;
    int c = (value & 0x4) == 0 ? 0 : 1;
    int d = (value & 0x8) == 0 ? 0 : 1;

    int e = (value & 0x10) == 0 ? 0 : 1;
    int f = (value & 0x20) == 0 ? 0 : 1;
    int g = (value & 0x40) == 0 ? 0 : 1;
    int h = (value & 0x80) == 0 ? 0 : 1;

    return a + b + c + d + e + f + g + h;
}

可以看到， 每运算一个位都有一个 if 判断，而要命的是这个 if 判断的结果是不稳定的，随机性极大。我写了一个计时程序，在我计算机中需要 12秒。（i5 6500 Release .net core 2.0 )

static void GetBitCountTest()
{
    var wathch = Stopwatch.StartNew();
            
    var rand = new Random();

    for (int i = 0; i < 10000_0000; i++)
    {
        int value = rand.Next();
        int p = getInt32TrueCount(value);
    }

    wathch.Stop();
    Console.WriteLine("GetBitCount 耗时：" + wathch.Elapsed.ToString());

}

 

方案2

第二种方法就是将 if 判断改为数学运算，方法是将 and 运算后的位 移动到 0位，这样就是 0 或 1 了。

static int getInt32TrueCount2(int value) {
    if (value == 0) {
        return 0;
    }

    return getByteTrueCount2(value & 0xFF) +
        getByteTrueCount2((value >> 8) & 0xFF) +
        getByteTrueCount2((value >> 16) & 0xFF) +
        getByteTrueCount2((value >> 24) & 0xFF);
}

static int getByteTrueCount2(int value) {
    return (value & 0x1) +
            ((value & 0x2) >> 1) +
            ((value & 0x4) >> 2) +
            ((value & 0x8) >> 3) +

            ((value & 0x10) >> 4) +
            ((value & 0x20) >> 5) +
            ((value & 0x40) >> 6) +
            ((value & 0x80) >> 7);
}

再次运行测试用例，执行时间提高到 2 秒！提高了6倍。

 

总结：

高性能计算时，避免使用 分支 指令，尽量使用数学运算符。