2023 day 1 (and presolve)

jesp1999 · jesp1999 · commit 69d0e29866c0 · 2023-12-01T00:48:05.000-05:00
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,3 @@
+.idea
+input.txt
+venv
diff --git a/2023/day1/main1.py b/2023/day1/main1.py
@@ -0,0 +1,32 @@
+def calval(s):
+    cv1, cv2 = 0, 0
+    for i in range(len(s)):
+        if s[i].isdigit():
+            cv1 = int(s[i])
+            break
+
+    for ii in range(len(s)):
+        i = len(s) - ii - 1
+        if s[i].isdigit():
+            cv2 = int(s[i])
+            break
+    return cv1 * 10 + cv2
+
+
+def solution(lines: list[str]):
+    ret = 0
+    for line in lines:
+        c = calval(line)
+        ret += c
+    return ret
+
+
+# data = '''1abc2
+# pqr3stu8vwx
+# a1b2c3d4e5f
+# treb7uchet
+# '''.split('\n')
+
+data = open('input.txt', 'r').read().split('\n')
+
+print(solution(data))
diff --git a/2023/day1/main2.py b/2023/day1/main2.py
@@ -0,0 +1,62 @@
+strnums = {'one': 1, 'two': 2, 'three': 3, 'four': 4, 'five': 5, 'six': 6,
+           'seven': 7, 'eight': 8, 'nine': 9}
+
+
+def substrnum(s):
+    for strnum in strnums:
+        if s.startswith(strnum):
+            return strnums[strnum]
+    return 0
+
+
+def calval(s):
+    cv1, cv2 = 0, 0
+    for i in range(len(s)):
+        if s[i].isdigit():
+            cv1 = int(s[i])
+            break
+        else:
+            c = substrnum(s[i:])
+            if c != 0:
+                cv1 = c
+                break
+
+    for ii in range(len(s)):
+        i = len(s) - ii - 1
+        if s[i].isdigit():
+            cv2 = int(s[i])
+            break
+        else:
+            c = substrnum(s[i:])
+            if c != 0:
+                cv2 = c
+                break
+    return cv1 * 10 + cv2
+
+
+def solution(lines: list[str]):
+    ret = 0
+    for line in lines:
+        c = calval(line)
+        ret += c
+    return ret
+
+
+# data = '''1abc2
+# pqr3stu8vwx
+# a1b2c3d4e5f
+# treb7uchet
+# '''.split('\n')
+
+# data = '''two1nine
+# eightwothree
+# abcone2threexyz
+# xtwone3four
+# 4nineeightseven2
+# zoneight234
+# 7pqrstsixteen
+# '''.split('\n')
+
+data = open('input.txt', 'r').read().split('\n')
+
+print(solution(data))
diff --git a/presolve.py b/presolve.py
@@ -0,0 +1,59 @@
+import os.path
+import requests
+from bs4 import BeautifulSoup
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+YEAR = int(os.environ.get('YEAR', 2023))
+DAY = int(os.environ.get('DAY', 1))
+SESSION_COOKIE = str(os.environ.get('SESSION_COOKIE', ''))
+
+cookies = {'session': SESSION_COOKIE}
+input_response = requests.get(
+    f'https://adventofcode.com/{YEAR}/day/{DAY}/input', cookies=cookies
+)
+
+if not os.path.exists(str(YEAR)):
+    os.mkdir(str(YEAR))
+
+dirname = f'{YEAR}/day{DAY}'
+
+if not os.path.exists(dirname):
+    os.mkdir(dirname)
+
+response = requests.get(
+    f'https://adventofcode.com/{YEAR}/day/{DAY}', cookies=cookies
+)
+response.raise_for_status()
+soup = BeautifulSoup(response.text, 'html.parser')
+code_tags = soup.find_all('pre')
+code_content_list = [code_tag.get_text() for code_tag in code_tags]
+test_cases = [
+    "# data = '''" + '\n# '.join(code_content.split("\n")) + r"'''.split('\n')"
+    for code_content in code_content_list
+]
+main_content = r'''...
+
+
+def solution(lines: list[str]):
+    ret = 0
+    ...
+    return ret
+
+
+''' + '\n\n'.join(test_cases) + r'''
+
+data = open('input.txt', 'r').read().split('\n')
+
+print(solution(data))
+'''
+
+main_path = f'{dirname}/main{len(code_content_list)}.py'
+if not os.path.exists(main_path):
+    with open(main_path, 'w+') as f:
+        f.write(main_content)
+
+with open(f'{dirname}/input.txt', 'w+') as f:
+    f.write(input_response.content.decode().strip('\n'))
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,3 @@
+beautifulsoup4
+python-dotenv~=1.0.0
+requests~=2.31.0

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+beautifulsoup4`
	`2`	`+python-dotenv~=1.0.0`
	`3`	`+requests~=2.31.0`