aoc

advent of code
git clone git://source.orangerot.dev:/aoc.git
Log | Files | Refs

main3.py (1041B)


      1 import sys
      2 
      3 def isVisible(lines, row, col):
      4     scenic = 1
      5     score = [1,2,3,4]
      6     score[0] = [x[col] < lines[row][col] for x in lines[:row][::-1]]
      7     score[1] = [x[col] < lines[row][col] for x in lines[row+1:]]
      8     score[2] = [x < lines[row][col] for x in lines[row][:col][::-1]]
      9     score[3] = [x < lines[row][col] for x in lines[row][col+1:]]
     10 
     11 
     12     for a in score:
     13         # print(a)
     14         dist = 0
     15         if (len(a) == 0):
     16             dist = 0
     17         elif (all(a)):
     18             dist = len(a)
     19         else:
     20             dist = a.index(False) + 1
     21         # print(dist)
     22         scenic *= dist
     23 
     24     print(row, col, "|", scenic)
     25     return scenic
     26 
     27 with open(sys.argv[1], "r") as file:
     28     lines = file.readlines()
     29     for a in range(len(lines)):
     30         lines[a] = lines[a].rstrip()
     31         # a = a.rstrip()
     32 
     33     best=0
     34     for row in range(len(lines)):
     35         for col in range(len(lines[0])):
     36             # print(lines[row][col], end="")
     37             yes = isVisible(lines, row, col)
     38             best = max(best, yes)
     39 
     40     print(best)