diff --git a/04/d.py b/04/d.py index 58c4d55..45d0d0b 100644 --- a/04/d.py +++ b/04/d.py @@ -7,20 +7,11 @@ tot1 = 0 tot2 = 0 for p in pairs: - a = p.split(",") + a = p.split(',')[0].split('-') + b = p.split(',')[1].split('-') - arrays = [] - - for i in range(0, len(a), 1): - b = a[i].split("-") - - if b[0] == b[1]: - arrays.append([int(b[0])]) - continue - arrays.append(range(int(b[0]), int(b[1])+1)) - - set_a = set(arrays[0]) - set_b = set(arrays[1]) + set_a = set(range(int(a[0]), int(a[1])+1)) + set_b = set(range(int(b[0]), int(b[1])+1)) if (set_a.issubset(set_b) or set_b.issubset(set_a)): tot1 += 1