2022 day 7: clean bash code & remove file size info (50% speed)

2022/day07/aoc.bash

@@ -13,31 +13,18 @@
 
 . common.bash
 
-declare -A files=()                               # list of files in dir
 declare -A sizes=()                               # file size
 declare -A dirs=()                                # directories
 declare -a lines                                  # shell lines
 declare -i curline=0                              # current line in shell
 declare curdir
 
-print_sizes() {
-    local idx
-    echo +++++++++++++++++++
-    for idx in "${!sizes[@]}"; do
-        printf "size(%s)=%d\n" "$idx" "${sizes[$idx]}"
-    done
-    echo +++++++++++++++++++
-}
-
 do_cd() {
     local dir="$1"
-    echo ">>> $curline ${lines[$curline]}"
     case "$dir" in
         "..")
-            echo ".. before: $curdir"
             curdir=${curdir%/*}
             [[ -z $curdir ]] && curdir="/"
-            echo ".. after: $curdir"
             ;;
         "/")
             curdir="/"
@@ -47,37 +34,25 @@ do_cd() {
             curdir+="$dir"
     esac
     dirs[$curdir]="$curdir"
-    echo "> CD $dir newdir=$curdir"
 }
 
 do_ls() {
-    local info file subdir remain
-    echo ">>> $curline ${lines[$curline]}"
+    local info file remain
     ((curline++))
     while [[ $curline -lt ${#lines[@]} && ${lines[$curline]:0:1} != "\$" ]]; do
         read -r info file <<< "${lines[$curline]}"
-        files[$curdir]+="$file"
         if [[ $info != dir ]]; then          # file
             remain="$curdir/$file"
             remain=${remain//+(\/)/\/}
-            #sizes[$fullname]=$info
-            # recurse up curdir and adjust sizes
-            #subdir=${curdir}
-            echo "info=$info [$remain]"
-            while [[ -n $remain ]]; do
-                echo "remain=$remain subdir=$subdir"
+            while [[ -n $remain ]]; do        # recurse up curdir and adjust sizes
                 ((sizes[$remain] += info))
                 remain=${remain%/*}
-                subdir=${remain##*/}
-                echo "  -> remain=$remain subdir=$subdir"
             done
             (( sizes["/"] += info ))
         fi
-        echo "ls: ${lines[$curline]}"
         ((curline++))
     done
     ((curline--))
-    print_sizes
 }
 
 parse() {
@@ -87,10 +62,6 @@ parse() {
 
     while ((curline < ${#lines[@]})); do
         read -ra line <<< "${lines[$curline]}"
-        if [[ "${line[0]}" != "\$" ]]; then
-            printf "ERROR line %d = %s\n" "$curline" "${lines[$curline]}"
-            exit 1
-        fi
         case "${line[1]}" in
             "cd")
                 do_cd "${line[2]}"
@@ -100,7 +71,6 @@ parse() {
                 ;;
         esac
         ((curline++))
-        printf "WARNING curline=%d\n lines=%d\n" "$curline" "${#lines[@]}"
     done
 }
 
@@ -111,23 +81,16 @@ solve() {
 
     if ((part == 1)); then
         for dir in "${dirs[@]}"; do
-            printf "size(%s)=%d\n" "$dir" "${sizes[$dir]}"
-            if ((sizes[$dir] <= 100000 )); then
-                ((res+=sizes[$dir]))
-            fi
+            ((sizes[$dir] <= 100000 )) && ((res+=sizes[$dir]))
         done
     else
+        (( res = sizes["/"] ))
         (( needed = sizes["/"] - (70000000-30000000) ))
-        printf "remain=%d\n" "$needed"
-        ((res = sizes["/"]))
         for dir in "${!dirs[@]}"; do
-            #printf "size(%s)=%d\n" "$dir" "${sizes[$dir]}"
             if (( sizes[$dir] >= needed )); then
-                printf "dir %s (%d) will free enough res=%d\n" "$dir" "${sizes[$dir]}" "$res"
                 if (( sizes[$dir] <= res )); then
                     mindir=$dir
                     ((res = sizes[$mindir]))
-                    printf "new mindir=%s (%d)\n" "$mindir" "$res"
                 fi
             fi
         done