40 lines
757 B
Python
40 lines
757 B
Python
def solve(data):
|
|
S1 = {
|
|
"A X": 1 + 3,
|
|
"A Y": 2 + 6,
|
|
"A Z": 3 + 0,
|
|
"B X": 1 + 0,
|
|
"B Y": 2 + 3,
|
|
"B Z": 3 + 6,
|
|
"C X": 1 + 6,
|
|
"C Y": 2 + 0,
|
|
"C Z": 3 + 3,
|
|
}
|
|
S2 = {
|
|
"A X": 3 + 0,
|
|
"A Y": 1 + 3,
|
|
"A Z": 2 + 6,
|
|
"B X": 1 + 0,
|
|
"B Y": 2 + 3,
|
|
"B Z": 3 + 6,
|
|
"C X": 2 + 0,
|
|
"C Y": 3 + 3,
|
|
"C Z": 1 + 6,
|
|
}
|
|
R = data.splitlines()
|
|
p1 = sum(S1[r] for r in R)
|
|
p2 = sum(S2[r] for r in R)
|
|
return p1, p2
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with open("./input/02.txt", "r") as f:
|
|
inp = f.read().strip()
|
|
|
|
p1, p2 = solve(inp)
|
|
|
|
print(p1)
|
|
print(p2)
|
|
|
|
assert p1 == 12772
|
|
assert p2 == 11618
|