mirror of
https://github.com/tanishq-dubey/aoc2023.git
synced 2025-01-01 10:32:34 -05:00
90 lines
1.7 KiB
Python
90 lines
1.7 KiB
Python
|
|
|
|
data = []
|
|
with open('input') as f:
|
|
data = f.readlines()
|
|
|
|
|
|
total = 0
|
|
|
|
for d in data:
|
|
dstring = ''.join(filter(str.isdigit, d))
|
|
if len(dstring) < 1:
|
|
continue
|
|
d1 = dstring[0]
|
|
d2 = dstring[-1]
|
|
total = total + int(f"{d1}{d2}")
|
|
|
|
|
|
print("=======")
|
|
print(total)
|
|
print("=======")
|
|
|
|
|
|
|
|
# Part 2
|
|
total = 0
|
|
|
|
dtt = {
|
|
"one": "o1e",
|
|
"two": "t2o",
|
|
"three": "t3e",
|
|
"four": "f4r",
|
|
"five": "f5e",
|
|
"six": "s6x",
|
|
"seven": "s7n",
|
|
"eight": "e8t",
|
|
"nine": "n9n",
|
|
}
|
|
|
|
for d in data:
|
|
tmp = d
|
|
|
|
for _, _ in dtt.items():
|
|
minidx = 10000000000
|
|
minval = "invalid"
|
|
for k,v in dtt.items():
|
|
try:
|
|
if tmp.index(k) < minidx:
|
|
minidx = tmp.index(k)
|
|
minval = k
|
|
except ValueError as e:
|
|
if "substring not found" in str(e):
|
|
continue
|
|
if minval != "invalid":
|
|
tmp = tmp.replace(minval, str(dtt[minval]), 1)
|
|
|
|
dstring = ''.join(filter(str.isdigit, tmp))
|
|
if len(dstring) < 1:
|
|
continue
|
|
d1 = dstring[0]
|
|
d2 = dstring[-1]
|
|
#print(d1, "-", d2, "----", tmp.strip(), d.strip())
|
|
total = total + int(f"{d1}{d2}")
|
|
|
|
|
|
print("======= PT 2")
|
|
print(total)
|
|
print("=======")
|
|
|
|
|
|
total = 0
|
|
for d in data:
|
|
tmp = d
|
|
for k, v in dtt.items():
|
|
try:
|
|
tmp = tmp.replace(k, str(v))
|
|
except Exception as e:
|
|
pass
|
|
dstring = ''.join(filter(str.isdigit, tmp))
|
|
if len(dstring) < 1:
|
|
continue
|
|
d1 = dstring[0]
|
|
d2 = dstring[-1]
|
|
#print(d1, "-", d2, "----", tmp.strip(), d.strip())
|
|
total = total + int(f"{d1}{d2}")
|
|
|
|
print("======= PT 2 - 2")
|
|
print(total)
|
|
print("=======")
|