yanick/adventofcode
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

refactoring

Browse Source
This commit is contained in:
Yanick Champoux 2021-12-10 12:23:45 -05:00
parent 85a9eb3107
commit eee25031e3
1 changed files with 17 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
2021/09/part2.mjs
View File

@ -2,43 +2,35 @@ import fs from "fs-extra";
import fp from "lodash/fp.js"; import fp from "lodash/fp.js";
import _ from "lodash"; import _ from "lodash";
import * as p1 from './part1.mjs'; import * as p1 from "./part1.mjs";
function getBasinSize( [x,y], grid, seen ) { function getBasinSize([x, y], grid, seen) {
if( seen[x][y] ) return 0; if (seen[x][y]) return 0;
seen[x][y] = true; seen[x][y] = true;
if( grid[x][y] == 9 ) return 0; if (grid[x][y] == 9) return 0;
let size = 1;
const neighbours = p1.genNeighbours(grid.length,grid[0].length); const neighbours = p1.genNeighbours(grid.length, grid[0].length);
const coords = p1.genCoords(grid);
neighbours(x,y).map( n => getBasinSize(n,grid,seen) ).forEach( return _.sum([
x => size+=x 1,
); ...neighbours(x, y).map((n) => getBasinSize(n, grid, seen)),
]);
return size;
} }
export function solution(grid) { export function solution(grid) {
const seen = grid.map( () => [] ); const seen = grid.map(() => []);
let basinSizes = []; let basinSizes = [];
for (const x of _.range(grid.length)) {
for( const x of _.range(grid.length) ) { for (const y of _.range(grid[x].length)) {
for( const y of _.range(grid[x].length) ) { basinSizes.push(getBasinSize([x, y], grid, seen));
if( seen[x][y] ) continue;
if( grid[x][y] == 9 ) {
seen[x][y] = true;
continue;
}
basinSizes.push( getBasinSize([x,y],grid,seen));
} }
} }
basinSizes.sort( (a,b) => a < b ? 1 : -1 ); return basinSizes
return basinSizes.splice(0,3).reduce( (a,b) => a*b ) ; .sort((a, b) => (a > b ? 1 : -1))
.splice(-3)
.reduce(_.multiply);
} }