chain is weird

This commit is contained in:
=
2024-02-06 17:35:56 +01:00
parent 7f6ac272c3
commit 966f72172f

View File

@@ -27,7 +27,7 @@ class PolyCube:
for last_addition in self.last_additions: for last_addition in self.last_additions:
growth_candidates = growth_candidates.union(generate_neighbors(*last_addition)) growth_candidates = growth_candidates.union(generate_neighbors(*last_addition))
growth_candidates = growth_candidates.difference(self.cubes) growth_candidates = growth_candidates.difference(self.cubes)
realizable_growth = chain(combinations(growth_candidates, i) for i in range(1, max_growth+1)) realizable_growth = chain.from_iterable(combinations(growth_candidates, i) for i in range(1, max_growth+1))
for addition in realizable_growth: for addition in realizable_growth:
yield PolyCube(self.cubes.union(addition), addition) yield PolyCube(self.cubes.union(addition), addition)