commit d46a294ac7e0761bc6be8e15b9114f0234377983 · kot.pink/aoc

+23

2.py

···

       1
       1
       +
       import sys

     

       2
       2
       +
       import math

     

       3
       3
       +
       

     

       4
       4
       +
       repeated_twice = 0

     

       5
       5
       +
       repeated_n = 0

     

       6
       6
       +
       file = '2.input' if len(sys.argv) <= 1 else sys.argv[1]

     

       7
       7
       +
       for pair in open(file).readline().split(","):

     

       8
       8
       +
           start, end = pair.split("-")

     

       9
       9
       +
           for i in range(int(start), int(end) + 1):

     

       10
       10
       +
               n = int(math.log10(i)) + 1

     

       11
       11
       +
               for length in range(n // 2, 0, -1):

     

       12
       12
       +
                   div, rem = divmod(n, length)

     

       13
       13
       +
                   if rem != 0:

     

       14
       14
       +
                       continue

     

       15
       15
       +
                   repeated = div * str(i)[:length]

     

       16
       16
       +
                   if int(repeated) == i:

     

       17
       17
       +
                       repeated_n += i

     

       18
       18
       +
                       if div == 2:

     

       19
       19
       +
                           repeated_twice += i

     

       20
       20
       +
                       break

     

       21
       21
       +
       

     

       22
       22
       +
       print(f'p1: {repeated_twice}')

     

       23
       23
       +
       print(f'p2: {repeated_n}')

2.test

···

       1
       1
       +
       11-22,95-115,998-1012,1188511880-1188511890,222220-222224,1698522-1698528,446443-446449,38593856-38593862,565653-565659,824824821-824824827,2121212118-2121212124