advent of code - day 1

day1.go

package aoc

func CalcFloor(s string) int {
	res := 0

	for i := 0; i < len(s); i++ {
		if s[i] == '(' {
			res++
		} else if s[i] == ')' {
			res--
		}
	}

	return res
}

func CalcCellarStep(s string) int {
	res := 0

	for i := 0; i < len(s); i++ {
		if s[i] == '(' {
			res++
		} else if s[i] == ')' {
			res--
		}

		if res < 0 {
			return i + 1
		}
	}

	return 0
}

day1_test.go

package aoc

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestCalcFloor(t *testing.T) {

	assert.Equal(t, 0, CalcFloor("(())"))
	assert.Equal(t, 0, CalcFloor("()()"))

	assert.Equal(t, 3, CalcFloor("((("))

	assert.Equal(t, 3, CalcFloor("(()(()("))

	assert.Equal(t, 3, CalcFloor("))((((("))

	assert.Equal(t, -1, CalcFloor("())"))
	assert.Equal(t, -1, CalcFloor("))("))

	assert.Equal(t, -3, CalcFloor(")))"))
	assert.Equal(t, -3, CalcFloor(")())())"))
}

func TestCalcCellarStep(t *testing.T) {
	assert.Equal(t, 1, CalcCellarStep(")"))
	assert.Equal(t, 5, CalcCellarStep("()())"))
}