AdventOfCode/2023/day7/p2_1.py

48 lines
899 B
Python

import sys
from pprint import pp
from collections import Counter
input_f = sys.argv[1]
cards = {}
with open(input_f) as file:
for line in file:
tmp = line.split()
cards[tmp[0]] = int(tmp[1])
def value(card):
return 'AKQJT98765432'.index(card)
def score(card):
tmp = sorted(list(Counter(card).values()),reverse=True)
if tmp == [5]:
return 1
elif tmp == [4,1]:
return 2
elif tmp == [3,2]:
return 3
elif tmp == [3,1,1]:
return 4
elif tmp == [2,2,1]:
return 5
elif tmp == [2,1,1,1]:
return 6
elif tmp == [1,1,1,1,1]:
return 7
scores = []
tmp = []
for i in cards:
tmp.append(tuple((score(i),tuple(value(c) for c in i),i)))
tmp = sorted(tmp, key=lambda x: (x[0], x[1]),reverse=True)
result = 0
for idx,i in enumerate(tmp):
result += cards[i[2]]*(idx+1)
print()
print(result)