Commit 4aa8854

committed

simplified

1 parent 8c998a6 commit 4aa8854Copy full SHA for 4aa8854

File tree

+11

-18

lines changed

+11

-18

lines changed

Lines changed: 11 additions & 18 deletions

Original file line number	Diff line number	Diff line change
`@@ -1,8 +1,4 @@`
`1`	`1`	`#!/usr/bin/env python`
`2`		`-from math import prod`
`3`		`-from functools import cmp_to_key`
`4`		`-`
`5`		`-`
`6`	`2`	`def cmp(left, right):`
`7`	`3`	`match left, right:`
`8`	`4`	`case int(), list():`
`@@ -18,22 +14,19 @@ def cmp(left, right):`
`18`	`14`	`return cmp(len(left), len(right))`
`19`	`15`
`20`	`16`
`21`		`-# fmt:off`
`22`		`-packets = [`
	`17`	`+# fmt:off`
	`18`	`+s1, two, six, packets =0, 1, 2, [`
`23`	`19`	`eval(line)`
`24`	`20`	`for line in open(0).read().splitlines()`
`25`	`21`	`if len(line)`
`26`	`22`	`]`
	`23`	`+# fmt:on`
	`24`	`+for i, (left, right) in enumerate(zip(packets[::2], packets[1::2])):`
	`25`	`+ if cmp(left, right) <= 0:`
	`26`	`+ s1 += i + 1`
	`27`	`+ for x in left, right:`
	`28`	`+ two += cmp(x, [[2]]) <= 0`
	`29`	`+ six += cmp(x, [[6]]) <= 0`
`27`	`30`
`28`		`-print(sum(`
`29`		`- i + 1`
`30`		`- for i, (left, right) in enumerate(zip(packets[::2], packets[1::2]))`
`31`		`- if cmp(left, right) <= 0`
`32`		`-))`
`33`		`-`
`34`		`-two_six = [[[2]], [[6]]]`
`35`		`-print(prod(`
`36`		`- i + 1`
`37`		`- for i, packet in enumerate(sorted(packets + two_six, key=cmp_to_key(cmp)))`
`38`		`- if packet in two_six`
`39`		`-))`
	`31`	`+print(s1)`
	`32`	`+print(two * six)`

Comments

(0)