Initial Commit - days 01-12

day07/ex2.bash

@@ -0,0 +1,69 @@
+#!/bin/bash
+#
+# ex1.bash: Advent2020 game, day 7/game 2.
+
+CMD=${0##*/}
+shopt -s extglob
+
+declare -A CONT
+
+function contains() {
+	local -i i size count=0 count2
+	local -i nbags
+	local key="$1" color
+	local -a contents
+
+	read -ra contents <<< "${CONT[$key]}"
+	((size = ${#contents[@]}/2))
+
+	for ((i=0; i<size; ++i)); do
+		nbags=${contents[i*2]}
+		color=${contents[i*2+1]}
+		((count+=nbags))
+		count2=$(contains "$color")
+		((count+=count2*nbags))
+	done
+	echo "$count"
+	return
+}
+
+function parse() {
+	local -a val=("$@")
+	local container="" in=""
+	local -i i=0 num=0
+	while [[ ! "${val[$i]}" =~ bag[s]* ]]; do
+		container+="${val[$i]}"
+		((i++))
+	done
+	CONT[$container]=""
+	((i++))
+	while [[ -n "${val[$i]}" ]]; do
+		case "${val[$i]}" in
+			contain)
+				;;
+			[[:digit:]]*)
+				num=${val[$i]}
+				;;
+			bag*)
+				CONT[$container]+="$num "
+				CONT[$container]+="$in "
+				in=""
+				num=0
+				;;
+			*)
+				in+="${val[$i]}"
+				;;
+		esac
+		((i++))
+	done
+}
+
+declare -a array
+while read -r -a array; do
+	parse "${array[@]}"
+done
+target="shinygold"
+res=$(contains $target)
+printf "%s : target=%s res=%d\n" "$CMD" "$target" "$res"
+
+exit 0