Working part 2
This commit is contained in:
290
2024/day-15/solution-2.sh
Normal file
290
2024/day-15/solution-2.sh
Normal file
@ -0,0 +1,290 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
|
||||||
|
FUNCNEST=99999
|
||||||
|
MAP_FILE=input-map
|
||||||
|
DIRECTIONS_FILE=input-movements
|
||||||
|
MAP_WIDTH=$(( ( $( head -1 "$MAP_FILE" | wc -c ) -1 ) * 2 ))
|
||||||
|
MAP_HEIGHT=$( < "$MAP_FILE" wc -l )
|
||||||
|
|
||||||
|
# Move the robot
|
||||||
|
move_robot() {
|
||||||
|
MAP_ARRAY[ROBOT_POSITION]=.
|
||||||
|
ROBOT_POSITION=$NEXT_POSITION
|
||||||
|
MAP_ARRAY[ROBOT_POSITION]=@
|
||||||
|
}
|
||||||
|
|
||||||
|
# Function to move box in n direction
|
||||||
|
vertical_move_box () {
|
||||||
|
local BOX_COORDINATE=$1
|
||||||
|
local BOX_MOVE_DIRECTION=$2
|
||||||
|
|
||||||
|
# Normalise coordinates
|
||||||
|
if [[ ${MAP_ARRAY[$BOX_COORDINATE]} == "[" ]]
|
||||||
|
then
|
||||||
|
local L_SIDE=$BOX_COORDINATE
|
||||||
|
local R_SIDE=$(( BOX_COORDINATE +1 ))
|
||||||
|
elif [[ ${MAP_ARRAY[$BOX_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
local L_SIDE=$(( BOX_COORDINATE -1 ))
|
||||||
|
local R_SIDE=$BOX_COORDINATE
|
||||||
|
fi
|
||||||
|
local L_SIDE_MOVE_COORDINATE=$(( L_SIDE + BOX_MOVE_DIRECTION ))
|
||||||
|
local R_SIDE_MOVE_COORDINATE=$(( R_SIDE + BOX_MOVE_DIRECTION ))
|
||||||
|
|
||||||
|
# Recurse if there are more boxes
|
||||||
|
if [[ ${MAP_ARRAY[L_SIDE_MOVE_COORDINATE]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[L_SIDE_MOVE_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
vertical_move_box "$L_SIDE_MOVE_COORDINATE" "$BOX_MOVE_DIRECTION"
|
||||||
|
fi
|
||||||
|
if [[ ${MAP_ARRAY[R_SIDE_MOVE_COORDINATE]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[R_SIDE_MOVE_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
vertical_move_box "$R_SIDE_MOVE_COORDINATE" "$BOX_MOVE_DIRECTION"
|
||||||
|
fi
|
||||||
|
|
||||||
|
MAP_ARRAY[L_SIDE_MOVE_COORDINATE]="["
|
||||||
|
MAP_ARRAY[R_SIDE_MOVE_COORDINATE]="]"
|
||||||
|
MAP_ARRAY[L_SIDE]="."
|
||||||
|
MAP_ARRAY[R_SIDE]="."
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Function to check if box can be moved in n direction
|
||||||
|
vertical_check_move_box () {
|
||||||
|
local BOX_COORDINATE=$1
|
||||||
|
local BOX_MOVE_DIRECTION=$2
|
||||||
|
|
||||||
|
# Normalise coordinates
|
||||||
|
if [[ ${MAP_ARRAY[$BOX_COORDINATE]} == "[" ]]
|
||||||
|
then
|
||||||
|
local L_SIDE=$BOX_COORDINATE
|
||||||
|
local R_SIDE=$(( BOX_COORDINATE +1 ))
|
||||||
|
elif [[ ${MAP_ARRAY[$BOX_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
L_SIDE=$(( BOX_COORDINATE -1 ))
|
||||||
|
R_SIDE=$BOX_COORDINATE
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Check if boxes are movable
|
||||||
|
local L_SIDE_MOVE_COORDINATE=$(( L_SIDE + BOX_MOVE_DIRECTION ))
|
||||||
|
local R_SIDE_MOVE_COORDINATE=$(( R_SIDE + BOX_MOVE_DIRECTION ))
|
||||||
|
if [[ ${MAP_ARRAY[$L_SIDE_MOVE_COORDINATE]} == '.' ]] && \
|
||||||
|
[[ ${MAP_ARRAY[$R_SIDE_MOVE_COORDINATE]} == '.' ]] # Can move
|
||||||
|
then
|
||||||
|
return 0
|
||||||
|
elif [[ ${MAP_ARRAY[$L_SIDE_MOVE_COORDINATE]} == '#' ]] || \
|
||||||
|
[[ ${MAP_ARRAY[$R_SIDE_MOVE_COORDINATE]} == '#' ]] # Blocked by wall
|
||||||
|
then
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Recurse if there are more boxes
|
||||||
|
if [[ ${MAP_ARRAY[L_SIDE_MOVE_COORDINATE]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[L_SIDE_MOVE_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
vertical_check_move_box "$L_SIDE_MOVE_COORDINATE" "$BOX_MOVE_DIRECTION"
|
||||||
|
if [[ $? -eq 1 ]]
|
||||||
|
then
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
if [[ ${MAP_ARRAY[R_SIDE_MOVE_COORDINATE]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[R_SIDE_MOVE_COORDINATE]} == "]" ]]
|
||||||
|
then
|
||||||
|
vertical_check_move_box "$R_SIDE_MOVE_COORDINATE" "$BOX_MOVE_DIRECTION"
|
||||||
|
if [[ $? -eq 1 ]]
|
||||||
|
then
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
horizontal_move_box () {
|
||||||
|
|
||||||
|
local BOX_COOR=$1
|
||||||
|
local BOX_MOVE_DIRECTION=$2
|
||||||
|
|
||||||
|
local NEXT_SPOT=$(( BOX_COOR + BOX_MOVE_DIRECTION * 2 ))
|
||||||
|
if [[ ${MAP_ARRAY[$NEXT_SPOT]} == "#" ]]
|
||||||
|
then
|
||||||
|
return 1 # Failed to move box
|
||||||
|
elif [[ ${MAP_ARRAY[$NEXT_SPOT]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[$NEXT_SPOT]} == "]" ]]
|
||||||
|
then
|
||||||
|
horizontal_move_box "$NEXT_SPOT" "$BOX_MOVE_DIRECTION"
|
||||||
|
if [[ $? -eq 1 ]] ; then return 1 ; fi
|
||||||
|
elif [[ ${MAP_ARRAY[$NEXT_SPOT]} != "." ]]
|
||||||
|
then
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Move the box
|
||||||
|
if [[ $NEXT_SPOT -lt $(( BOX_COOR + BOX_MOVE_DIRECTION )) ]]
|
||||||
|
then
|
||||||
|
MAP_ARRAY[NEXT_SPOT]="["
|
||||||
|
MAP_ARRAY[BOX_COOR + BOX_MOVE_DIRECTION]="]"
|
||||||
|
else
|
||||||
|
MAP_ARRAY[BOX_COOR + BOX_MOVE_DIRECTION]="["
|
||||||
|
MAP_ARRAY[NEXT_SPOT]="]"
|
||||||
|
fi
|
||||||
|
MAP_ARRAY[BOX_COOR]="."
|
||||||
|
}
|
||||||
|
|
||||||
|
# Load map
|
||||||
|
read -r -a TEST_ARRAY <<< "$(
|
||||||
|
< "$MAP_FILE" paste -s -d "" |
|
||||||
|
sed -E '
|
||||||
|
s/#/##/g;
|
||||||
|
s/\./../g;
|
||||||
|
s/@/@./g;
|
||||||
|
s/O/[]/g;
|
||||||
|
s/(.)(.)/\1 \2 /g'
|
||||||
|
)"
|
||||||
|
read -r -a MAP_ARRAY <<< "$(
|
||||||
|
< "$MAP_FILE" paste -s -d "" |
|
||||||
|
sed -E '
|
||||||
|
s/#/##/g;
|
||||||
|
s/\./../g;
|
||||||
|
s/@/@./g;
|
||||||
|
s/O/[]/g;
|
||||||
|
s/(.)(.)/\1 \2 /g'
|
||||||
|
)"
|
||||||
|
MAP_LEN=${#MAP_ARRAY[@]}
|
||||||
|
|
||||||
|
print_map () {
|
||||||
|
printf "%s " "${MAP_ARRAY[@]}" |
|
||||||
|
fold -w $(( MAP_WIDTH * 2 ))
|
||||||
|
printf "\n"
|
||||||
|
}
|
||||||
|
|
||||||
|
# Count boxes
|
||||||
|
#BOX_COUNT=0
|
||||||
|
#for (( i=0; i<MAP_LEN; i++ ))
|
||||||
|
#do
|
||||||
|
# if [[ ${MAP_ARRAY[$i]} == "[" ]]
|
||||||
|
# then
|
||||||
|
# (( BOX_COUNT++ ))
|
||||||
|
# fi
|
||||||
|
#done
|
||||||
|
#printf "Box count: %s\n" "$BOX_COUNT"
|
||||||
|
|
||||||
|
# Get robot position
|
||||||
|
for (( i=0; i<MAP_LEN; i++ ))
|
||||||
|
do
|
||||||
|
if [[ ${MAP_ARRAY[$i]} == "@" ]]
|
||||||
|
then
|
||||||
|
ROBOT_POSITION=$i
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
printf "Robot position : %s\n" "$ROBOT_POSITION"
|
||||||
|
|
||||||
|
# Hardcode direction values
|
||||||
|
UP=$(( - MAP_WIDTH ))
|
||||||
|
DOWN=$MAP_WIDTH
|
||||||
|
LEFT=-1
|
||||||
|
RIGHT=1
|
||||||
|
|
||||||
|
# Iterate through directions
|
||||||
|
MOVES=0
|
||||||
|
print_map
|
||||||
|
while read -r DIRECTION
|
||||||
|
do
|
||||||
|
|
||||||
|
# Interpret directions
|
||||||
|
if [[ $DIRECTION == "^" ]]
|
||||||
|
then
|
||||||
|
DIRECTION_VALUE=$UP
|
||||||
|
elif [[ $DIRECTION == "v" ]]
|
||||||
|
then
|
||||||
|
DIRECTION_VALUE=$DOWN
|
||||||
|
elif [[ $DIRECTION == "<" ]]
|
||||||
|
then
|
||||||
|
DIRECTION_VALUE=$LEFT
|
||||||
|
elif [[ $DIRECTION == ">" ]]
|
||||||
|
then
|
||||||
|
DIRECTION_VALUE=$RIGHT
|
||||||
|
fi
|
||||||
|
|
||||||
|
NEXT_POSITION=$(( ROBOT_POSITION + DIRECTION_VALUE ))
|
||||||
|
|
||||||
|
# Box in front
|
||||||
|
if [[ ${MAP_ARRAY[$NEXT_POSITION]} == "[" ]] || \
|
||||||
|
[[ ${MAP_ARRAY[$NEXT_POSITION]} == "]" ]]
|
||||||
|
then
|
||||||
|
#printf "Hit box.\n"
|
||||||
|
if [[ $DIRECTION_VALUE -eq $LEFT ]] || \
|
||||||
|
[[ $DIRECTION_VALUE -eq $RIGHT ]]
|
||||||
|
then
|
||||||
|
#printf "Left right box movement.\n"
|
||||||
|
if horizontal_move_box "$NEXT_POSITION" "$DIRECTION_VALUE"
|
||||||
|
then
|
||||||
|
move_robot
|
||||||
|
fi
|
||||||
|
elif [[ $DIRECTION_VALUE -eq $UP ]] || \
|
||||||
|
[[ $DIRECTION_VALUE -eq $DOWN ]]
|
||||||
|
then
|
||||||
|
#printf "Up down box movement.\n"
|
||||||
|
if vertical_check_move_box "$NEXT_POSITION" "$DIRECTION_VALUE"
|
||||||
|
then
|
||||||
|
vertical_move_box "$NEXT_POSITION" "$DIRECTION_VALUE"
|
||||||
|
move_robot
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Wall in front, do nothing
|
||||||
|
elif [[ ${MAP_ARRAY[$NEXT_POSITION]} == "#" ]]
|
||||||
|
then
|
||||||
|
#printf "Next: %s Skipping\n" "${MAP_ARRAY[$NEXT_POSITION]}"
|
||||||
|
:
|
||||||
|
# Empty space, move forward
|
||||||
|
elif [[ ${MAP_ARRAY[$NEXT_POSITION]} == "." ]]
|
||||||
|
then
|
||||||
|
#printf "Next: %s Move robot forward\n" "${MAP_ARRAY[$NEXT_POSITION]}"
|
||||||
|
move_robot
|
||||||
|
fi
|
||||||
|
(( MOVES++ ))
|
||||||
|
|
||||||
|
#print_map
|
||||||
|
done <<< "$( < "$DIRECTIONS_FILE" paste -s -d "" | sed -E 's/(.)(.)/\1 \2 /g' | tr ' ' '\n' )"
|
||||||
|
printf "Moves: %s\n" "$MOVES"
|
||||||
|
|
||||||
|
print_map
|
||||||
|
|
||||||
|
# Calculate GPS values for boxes
|
||||||
|
SUM=0
|
||||||
|
BOX_COUNT=0
|
||||||
|
for (( i=0; i<MAP_LEN; i++ ))
|
||||||
|
do
|
||||||
|
if [[ ${MAP_ARRAY[i]} == "[" ]]
|
||||||
|
then
|
||||||
|
(( BOX_COUNT++ ))
|
||||||
|
#X_L=$(( i % MAP_WIDTH ))
|
||||||
|
#X_R=$(( MAP_WIDTH - (i % MAP_WIDTH ) -1 ))
|
||||||
|
#Y_T=$(( i / MAP_WIDTH * 100 ))
|
||||||
|
#Y_B=$(( ( MAP_HEIGHT - i / MAP_WIDTH ) * 100 ))
|
||||||
|
#X=$(( X_R > X_L ? X_L : X_R ))
|
||||||
|
#Y=$(( Y_T > Y_B ? Y_B : Y_T ))
|
||||||
|
X=$(( i % MAP_WIDTH ))
|
||||||
|
Y=$(( i / MAP_WIDTH * 100 ))
|
||||||
|
GPS=$(( X + Y ))
|
||||||
|
#printf "Box %s X: %s Y: %s GPS: %s\n" "$i" "$X" "$Y" "$GPS"
|
||||||
|
#if [[ $GPS -lt 0 ]] ; then printf "Negative GPS: %s for box %s\n" "$GPS" "$i"; continue; fi
|
||||||
|
(( SUM+=GPS ))
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
printf "Sum: %s\n" "$SUM"
|
||||||
|
printf "Box count: %s\n" "$BOX_COUNT"
|
||||||
|
|
||||||
|
# Compare and test if walls have moved
|
||||||
|
for (( i=0; i<MAP_LEN; i++ ))
|
||||||
|
do
|
||||||
|
if [[ ${TEST_ARRAY[i]} == "#" ]]
|
||||||
|
then
|
||||||
|
if [[ ${TEST_ARRAY[i]} != "${MAP_ARRAY[i]}" ]]
|
||||||
|
then
|
||||||
|
printf "Index %s Wall changed.\n" "$i"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
done
|
||||||
Reference in New Issue
Block a user