From 4a18141f582733953fff0fe7a6ee116b6094c14d Mon Sep 17 00:00:00 2001
From: Trey Bastian <2991824+TreyBastian@users.noreply.github.com>
Date: Mon, 2 Dec 2024 19:12:25 +0000
Subject: formatting and part 2 done

---
 2024/day_02_part_2.f90 | 156 +++++++++++++++++++++++++------------------------
 1 file changed, 80 insertions(+), 76 deletions(-)

(limited to '2024/day_02_part_2.f90')

diff --git a/2024/day_02_part_2.f90 b/2024/day_02_part_2.f90
index affae98..13a0812 100644
--- a/2024/day_02_part_2.f90
+++ b/2024/day_02_part_2.f90
@@ -1,86 +1,90 @@
-module day2_utils 
-  implicit none
-  contains
-  subroutine has_fail_value(list, res) 
-    implicit none
-    integer, allocatable, intent(in) :: list(:)
-    integer :: i
-    integer, intent(out):: res
-    logical is_decrement
+module day2_utils
+   implicit none
+contains
+   subroutine is_safe(list, res)
+      implicit none
+      integer, allocatable, intent(in) :: list(:)
+      logical, intent(out):: res
+      integer :: i, diff
+      logical is_decrement
 
-    res = -1 
-    do i = 1, size(list) - 1
-    if (list(i) == list(i+1)) then 
-       res= i 
-       exit
-      end if
-      if(i == 1 .and. list(i) > list(i+1)) then
-        is_decrement = .true.
-      else if (i==1 .and. list(i) < list(i+1)) then
-        is_decrement = .false.
-      end if
-      
-      if(is_decrement)  then
-        if (list(i) < list(i+1) .or. abs(list(i) - list(i+1)) > 3) then 
-         res = i 
-          exit
-        end if
-      else
-        if (list(i) >  list(i+1) .or. abs(list(i) - list(i+1)) > 3) then 
-          res = i
-          exit
-        end if
+      res = .true.
+      is_decrement = .false.
+      do i = 1, size(list) - 1
+         if(i == 1 .and. list(i) > list(i+1)) then
+            is_decrement = .true.
+         end if
+         diff = abs(list(i) - list(i+1))
+         if(is_decrement)  then
+            if (list(i) < list(i+1) .or. diff > 3 .or. diff < 1) then
+               res = .false.
+               exit
+            end if
+         else
+            if (list(i) >  list(i+1) .or. diff > 3 .or. diff < 1) then
+               res = .false.
+               exit
+            end if
+         end if
+      end do
+   end subroutine is_safe
+   subroutine safe_with_dampener(list, res)
+      implicit none
+      integer, allocatable, intent(in) :: list(:)
+      integer, allocatable :: work(:)
+      logical, intent(out):: res
+      integer :: i, x, idx
+      res = .true.
+      call is_safe(list, res)
+      if(.not. res) then
+         allocate(work(size(list)-1))
+         do i = 1, size(list)
+            idx = 1
+            do x = 1, size(list)
+               if (x /= i) then
+                  work(idx) = list(x)
+                  idx = idx + 1
+               end if
+            end do
+            res = .true.
+            call is_safe(work, res)
+            if(res) then
+               exit
+            end if
+         end do
       end if
-    end do
-  end subroutine has_fail_value
+   end subroutine safe_with_dampener
 end module day2_utils
 
 program day_02_part_2
-  use day2_utils
-  implicit none
-  logical :: is_decrement 
-  integer :: io, i,x,y, fail, spaces, res
-  integer, dimension(:), allocatable :: list, list2
-  character(len=100) :: line
+   use day2_utils
+   implicit none
+   logical :: is_decrement, safe
+   integer :: io, i, x, spaces, res
+   integer, dimension(:), allocatable :: list
+   character(len=100) :: line
 
-  open(newunit=io, file='./day_02_input.txt', status='old', action='read')
-
-  res = 0
-  do i = 1, 1000 
-    read(io, '(a)') line ! reading the line 
-    spaces = 1 ! reset space count
-    do x = 1, len(trim(line))
-      ! counting spaces to see how big of an array we need to allocate
-      if(iachar(line(x:x))== 32 ) then
-        spaces = spaces+1 
-      end if
-    end do
+   open(newunit=io, file='./day_02_input.txt', status='old', action='read')
 
-    allocate(list(spaces))
-    read(line, *) list !reading the ints into an array
-    fail = -1
-    call has_fail_value(list, fail) 
-    if (fail > 0) then 
-      y = 1
-      allocate(list2(spaces -1))
-      do x = 1, size(list)
-        if(x /= fail) then
-          list2(y) = list(x)
-          y = y+1
-        end if
+   res = 0
+   do i = 1, 1000
+      read(io, '(a)') line ! reading the line
+      spaces = 1 ! reset space count
+      do x = 1, len(trim(line))
+         ! counting spaces to see how big of an array we need to allocate
+         if(iachar(line(x:x))== 32 ) then
+            spaces = spaces+1
+         end if
       end do
-      fail = -1
-      call has_fail_value(list2, fail)
-      if(fail > 0) then
-        print*, list2(:)
-       end if
-      deallocate(list2)
 
-     end if
-     if (fail == -1) then
-     res = res + 1 
-    endif
-    deallocate(list)
-  end do
-  print*, res 
+      allocate(list(spaces))
+      read(line, *) list !reading the ints into an array
+      safe = .true. !set fail to false by default
+      call safe_with_dampener(list, safe) ! check if the array has a fail idx
+      if (safe) then
+         res = res + 1
+      endif
+      deallocate(list)
+   end do
+   print*, res
 end program day_02_part_2
-- 
cgit v1.2.3