04_count_passphrases.go

package main

import (
	"fmt"
	"metalim/advent/2017/lib/source"
	"sort"

	. "github.com/logrusorgru/aurora"
)

func main() {
	var ins source.Inputs

	ins = ins.Test(1, `aa bb cc dd aa`, `0`)
	ins = ins.Test(1, `aa bb cc dd aaa`, `1`)
	ins = ins.Test(2, `abcde xyz ecdab`, `0`)

	for p := range ins.Advent(2017, 4) {
		fmt.Println(Brown("\n" + p.Name))
		ssw := p.Lines().Words()
		fmt.Println(len(ssw), "lines", Black(ssw[0]).Bold())

		if p.Part(1) {
			var count int
		LINES:
			for _, l := range ssw {
				words := map[string]bool{}
				for _, w := range l {
					if words[w] {
						continue LINES
					}
					words[w] = true
				}
				count++
			}
			p.SubmitInt(1, count)
		}

		if p.Part(2) {
			var count int
		LINES2:
			for _, l := range ssw {
				words := map[string]bool{}
				for _, w := range l {
					rs := []rune(w)
					sort.Slice(rs, func(i, j int) bool { return rs[i] < rs[j] })
					w = string(rs)
					if words[w] {
						continue LINES2
					}
					words[w] = true
				}
				count++
			}
			p.SubmitInt(2, count)
		}
	}
}