Archived
165 lines
3.0 KiB
Go
165 lines
3.0 KiB
Go
package three
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"log"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func Run() {
|
|
values := readInput()
|
|
heatmap := createHeatmap(values)
|
|
fmt.Printf("power: %d\n", calcPower(calcGamma(heatmap, len(values)), calcEpsilon(heatmap, len(values))))
|
|
fmt.Printf("oxygen * co2 (life support): %d\n", calcOxygen(values)*calcCO2(values))
|
|
}
|
|
|
|
func calcOxygen(values [][]int) int {
|
|
return calcOxygenOrCO2(values, false)
|
|
}
|
|
|
|
func calcCO2(values [][]int) int {
|
|
return calcOxygenOrCO2(values, true)
|
|
}
|
|
|
|
func calcOxygenOrCO2(values [][]int, getCO2 bool) int {
|
|
valueStore := make([][][]int, 0)
|
|
valueStore = append(valueStore, values)
|
|
for i := 0; i < len(values[0]); i++ {
|
|
commonValues := getCommonValues(valueStore[i], getCO2)
|
|
newValues := make([][]int, 0)
|
|
for _, v := range valueStore[i] {
|
|
if v[i] == commonValues[i] {
|
|
newValues = append(newValues, v)
|
|
}
|
|
}
|
|
|
|
valueStore = append(valueStore, newValues)
|
|
|
|
if len(newValues) == 1 {
|
|
break
|
|
}
|
|
}
|
|
|
|
return convertToDecimal(valueStore[len(valueStore)-1][0])
|
|
}
|
|
|
|
func getCommonValues(values [][]int, least bool) []int {
|
|
ones := make([]int, len(values[0]))
|
|
zeros := make([]int, len(values[0]))
|
|
result := make([]int, len(values[0]))
|
|
for _, v := range values {
|
|
for i, b := range v {
|
|
if b == 1 {
|
|
ones[i]++
|
|
} else if b == 0 {
|
|
zeros[i]++
|
|
}
|
|
}
|
|
}
|
|
for k := range result {
|
|
if ones[k] > zeros[k] {
|
|
if least {
|
|
result[k] = 0
|
|
} else {
|
|
result[k] = 1
|
|
}
|
|
} else {
|
|
if least {
|
|
result[k] = 1
|
|
} else {
|
|
result[k] = 0
|
|
}
|
|
}
|
|
if ones[k] == zeros[k] {
|
|
if least {
|
|
result[k] = 0
|
|
} else {
|
|
result[k] = 1
|
|
}
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func calcPower(gamma []int, epsilon []int) int {
|
|
return convertToDecimal(gamma) * convertToDecimal(epsilon)
|
|
}
|
|
|
|
func convertToDecimal(i []int) int {
|
|
s := []string{}
|
|
for _, v := range i {
|
|
s = append(s, strconv.Itoa(v))
|
|
}
|
|
r, _ := strconv.ParseInt(strings.Join(s, ""), 2, 64)
|
|
return int(r)
|
|
}
|
|
|
|
func calcGamma(heatmap []int, lines int) []int {
|
|
gamma := make([]int, len(heatmap))
|
|
for k, v := range heatmap {
|
|
if v > lines/2 {
|
|
gamma[k] = 1
|
|
} else {
|
|
gamma[k] = 0
|
|
}
|
|
}
|
|
return gamma
|
|
}
|
|
|
|
func calcEpsilon(heatmap []int, lines int) []int {
|
|
gamma := make([]int, len(heatmap))
|
|
for k, v := range heatmap {
|
|
if v < lines/2 {
|
|
gamma[k] = 1
|
|
} else {
|
|
gamma[k] = 0
|
|
}
|
|
}
|
|
return gamma
|
|
}
|
|
|
|
func createHeatmap(values [][]int) []int {
|
|
heatmap := make([]int, len(values[0]))
|
|
for _, v := range values {
|
|
for k2, v2 := range v {
|
|
if v2 == 1 {
|
|
heatmap[k2]++
|
|
}
|
|
}
|
|
}
|
|
return heatmap
|
|
}
|
|
|
|
func readInput() [][]int {
|
|
file, err := os.Open("./pkg/three/input.txt")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer file.Close()
|
|
values := make([][]int, 0)
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
|
|
for scanner.Scan() {
|
|
line := strings.Split(scanner.Text(), "")
|
|
|
|
val := make([]int, 0)
|
|
for _, v := range line {
|
|
x, err := strconv.Atoi(v)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
val = append(val, x)
|
|
}
|
|
values = append(values, val)
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return values
|
|
}
|