import { task } from '@alexaegis/advent-of-code-lib';
import { Direction, Vec2 } from '@alexaegis/advent-of-code-lib/model';
import packageJson from '../package.json';
import { getLowPoints } from './p1.js';

export const p2 = (input: string): number => {
	const heightMap = input.toVectorMap((s) => s.tryInt());
	const lowPoints = getLowPoints(heightMap);

	const flow = (coord: Vec2, checked = new Set<string>()): Vec2[] => {
		const neighbours = Direction.cardinalDirections
			.map((direction) => coord.add(direction))
			.filter((nc) => {
				const ncs = nc.toString();
				const nch = heightMap.get(ncs);
				return !checked.has(ncs) && nch !== 9 && nch !== undefined;
			});
		for (const neighbour of neighbours) checked.add(neighbour.toString());
		return [...neighbours, ...neighbours.flatMap((neighbour) => flow(neighbour, checked))];
	};

	const basins = lowPoints.map(([key]) => flow(new Vec2(key)));
	const largestThreeBasins = basins.sort((a, b) => b.length - a.length).slice(0, 3);
	return largestThreeBasins.map((basin) => basin.length).reduce((a, n) => a * n, 1);
};

await task(p2, packageJson.aoc); // 1047744 ~26.36ms