golang int64 排序_在 Golang 中针对 int64 类型优化 abs()

前言

Go 语言没有内置 abs() 标准函数来计算整数的绝对值，这里的绝对值是指负数、正数的非负表示。

我最近为了解决 Advent of Code 2017 上边的 Day 20 难题，自己实现了一个 abs() 函数。如果你想学点新东西或试试身手，可以去一探究竟。

Go 实际上已经在 math 包中实现了 abs() : math.Abs ，但对我的问题并不适用，因为它的输入输出的值类型都是 float64，我需要的是 int64。通过参数转换是可以使用的，不过将 float64 转为 int64 会产生一些开销，且转换值很大的数会发生截断，这两点都会在文章说清楚。

帖子 Pure Go math.Abs outperforms assembly version 讨论了针对浮点数如何优化 math.Abs，不过这些优化的方法因底层编码不同，不能直接应用在整型上。

类型转换 VS 分支控制的方法

对我来说取绝对值最简单的函数实现是：输入参数 n 大于等于 0 直接返回 n，小于零则返回 -n(负数取反为正)，这个取绝对值的函数依赖分支控制结构来计算绝对值，就命名为： abs.WithBranch

package abs

func WithBranch(n int64) int64 {

if n < 0 {

return -n

}

return n

}

成功返回 n 的绝对值，这就是 Go v1.9.x math.Abs 对 float64 取绝对值的实现。不过当进行类型转换(int64 to float64)再取绝对值时，1.9.x 是否做了改进？我们可以验证一下：

package abs

func WithStdLib(n int64) int64 {

return int64(math.Abs(float64(n)))

}

上边的代码中，将 n 先从 int64 转成 float64，通过 math.Abs 取到绝对值后再转回 int64，多次转换显然会造成性能开销。可以写一个基准测试来验证一下：

$ go test -bench=.

goos: darwin

goarch: amd64

pkg: github.com/cavaliercoder/abs

BenchmarkWithBranch-8 2000000000 0.30 ns/op

BenchmarkWithStdLib-8 2000000000 0.79 ns/op

PASS

ok github.com/cavaliercoder/abs 2.320s

测试结果：0