| 4 | 4 | // |
| 5 | 5 | |
| 6 | 6 | /** |
| 7 | | * @param {string[]} input |
| 8 | | * @returns {string[]} |
| 7 | * Annotates a flower field by replacing empty spaces with the count of adjacent flowers. |
| 8 | * @param {string[]} input - A 2D array represented as a list of strings, where each string is a row. |
| 9 | * '*' represents a flower, ' ' represents an empty space. |
| 10 | * @returns {string[]} - The annotated flower field with counts replacing empty spaces. |
| 9 | 11 | */ |
| 10 | 12 | export const annotate = (input) => { |
| 11 | | throw new Error('Remove this statement and implement this function'); |
| 13 | // Edge Case: Handle empty input |
| 14 | if (!input || input.length === 0) { |
| 15 | return []; |
| 16 | } |
| 17 | |
| 18 | const rows = input.length; |
| 19 | const cols = input[0].length; |
| 20 | |
| 21 | // Edge Case: Handle rows with inconsistent lengths |
| 22 | for (let i = 0; i < rows; i++) { |
| 23 | if (input[i].length !== cols) { |
| 24 | throw new Error('All rows must have the same length'); |
| 25 | } |
| 26 | } |
| 27 | |
| 28 | // Create a 2D array for the result |
| 29 | const result = Array(rows).fill().map(() => Array(cols).fill('')); |
| 30 | |
| 31 | // Define the 8 possible directions (horizontal, vertical, diagonal) |
| 32 | const directions = [ |
| 33 | [-1, -1], [-1, 0], [-1, 1], |
| 34 | [0, -1], [0, 1], |
| 35 | [1, -1], [1, 0], [1, 1] |
| 36 | ]; |
| 37 | |
| 38 | // Process each cell |
| 39 | for (let i = 0; i < rows; i++) { |
| 40 | for (let j = 0; j < cols; j++) { |
| 41 | // If the cell is a flower, keep it as is |
| 42 | if (input[i][j] === '*') { |
| 43 | result[i][j] = '*'; |
| 44 | } else { |
| 45 | // Count adjacent flowers |
| 46 | let count = 0; |
| 47 | |
| 48 | // Check all 8 directions |
| 49 | for (const [dx, dy] of directions) { |
| 50 | const ni = i + dx; |
| 51 | const nj = j + dy; |
| 52 | |
| 53 | // Edge Case: Check if the adjacent cell is within bounds |
| 54 | if (ni >= 0 && ni < rows && nj >= 0 && nj < cols) { |
| 55 | if (input[ni][nj] === '*') { |
| 56 | count++; |
| 57 | } |
| 58 | } |
| 59 | } |
| 60 | |
| 61 | // If no adjacent flowers, leave empty, otherwise put the count |
| 62 | result[i][j] = count === 0 ? ' ' : count.toString(); |
| 63 | } |
| 64 | } |
| 65 | } |
| 66 | |
| 67 | // Convert the 2D array back to an array of strings |
| 68 | return result.map(row => row.join('')); |
| 12 | 69 | }; |
| 70 | |
| 71 | // Handled Edge Cases: Empty input, inconsistent row lengths, out-of-bounds adjacent cells |