Cantor pairing function

cantor_pairing.go

// http://en.wikipedia.org/wiki/Pairing_function
package main

import (
	"fmt"
	"math"
)

func InvertedCantorPairing(z int) (int, int) {
	w := int(math.Floor((math.Sqrt(float64(8*z+1)) - 1) / 2))
	t := (w*w + w) / 2
	y := z - t
	x := w - y
	return x, y
}

func CantorPairing(k1, k2 int) int {
	return (k1+k2)*(k1+k2+1)/2 + k2
}

func main() {
	for i := 0; i < 100; i++ {
		x, y := InvertedCantorPairing(i)
		fmt.Println(i, x, y, CantorPairing(x, y))
	}
}