1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
module day_07_part_2_utils
use iso_fortran_env, only: int64
implicit none
contains
recursive function test_values(arr, target, next_idx, current_value) result(pass)
implicit none
integer(kind=int64), intent(in) :: target, current_value, arr(:)
integer, intent(in) :: next_idx
character(len=20) :: str_1, str_2, concat_str
integer(kind=int64) :: concat_value
logical :: pass
if(next_idx > size(arr)) then
pass = current_value == target
return
end if
write(str_1, '(I0)') current_value
write(str_2, '(I0)') arr(next_idx)
concat_str = trim(str_1) // trim(str_2)
read(concat_str, '(I20)') concat_value
pass = test_values(arr, target, next_idx + 1, current_value + arr(next_idx)) .or. &
test_values(arr, target, next_idx + 1, current_value * arr(next_idx)) .or. &
test_values(arr, target, next_idx + 1, concat_value)
end function test_values
logical function is_calibrated(arr, target)
implicit none
integer(kind=int64), intent(in) :: target, arr(:)
is_calibrated = test_values(arr, target, 2, arr(1))
end function is_calibrated
end module day_07_part_2_utils
program day_07_part_2
use iso_fortran_env, only: int64
use day_07_part_2_utils
implicit none
integer(kind=int64) :: test_number, res
integer(kind=int64), allocatable:: work(:)
integer :: io, ios, idx, i, ct
character(len=200) :: line, work_line
logical :: pass
res = 0
open(newunit=io, file='./day_07_input.txt', status='old', action='read')
do
read(io, '(a)', iostat=ios) line
if (ios /= 0) exit
idx = index(line, ':')
if (idx > 0) then
read(line(1:idx-1), *) test_number
ct = 0
work_line = trim(line(idx+1:len(trim(line))))
do i = 1 , len(trim(work_line))
if(work_line(i:i) == ' ') then
ct = ct + 1
end if
end do
allocate(work(ct))
read(work_line, *) work(:)
end if
pass = is_calibrated(work, test_number)
if(pass) then
res = res + test_number
end if
deallocate(work)
end do
print*, "Result: ", res
end program day_07_part_2
|
