diff --git a/main.js b/main.js index 771cbf0..ce1589a 100644 --- a/main.js +++ b/main.js @@ -4,9 +4,130 @@ // /** - * @param {string[]} input - * @return {string[]} + * Transposes input text by converting rows to columns and columns to rows. + * For inputs with rows of different lengths, pads with spaces as needed. + * @param {string[]} input - Array of strings representing rows of text + * @return {string[]} - Array of strings representing transposed text */ export function transpose(input) { - throw new Error('Remove this line and implement the function'); -}; + // Handle empty input + if (!input || input.length === 0) { + return []; + } + + // Handle single empty string + if (input.length === 1 && input[0] === '') { + return []; + } + + // Find the maximum length among all rows + const maxLength = Math.max(...input.map(row => row.length)); + + // Create result array + const result = []; + + // For each column position (0 to maxLength-1) + for (let col = 0; col < maxLength; col++) { + let newRow = ''; + + // For each row, get the character at this column position + for (let row = 0; row < input.length; row++) { + if (col < input[row].length) { + newRow += input[row][col]; + } else { + // Pad with space if this row is shorter + // But only if there are rows below that have content at this column + let hasContentBelow = false; + for (let checkRow = row + 1; checkRow < input.length; checkRow++) { + if (col < input[checkRow].length) { + hasContentBelow = true; + break; + } + } + + if (hasContentBelow) { + newRow += ' '; + } + } + } + + // Add the row to result + result.push(newRow); + } + + // Handle trailing spaces from input + // If the last few rows contain only spaces that come from the input (not padding), + // and those spaces are from the same input row, append them to the previous row + + // Work backwards from the end to find trailing space-only rows that should be combined + let trailingSpaceRows = 0; + + // Count how many trailing rows contain only spaces + for (let i = result.length - 1; i >= 0; i--) { + if (result[i].trim() === '') { + // This row contains only spaces + // Check if these spaces are from the input (not just padding) + const colIndex = i; + let allSpacesFromInput = true; + + for (let row = 0; row < input.length; row++) { + if (colIndex < input[row].length) { + if (input[row][colIndex] !== ' ') { + // This position has a non-space character, so this isn't just spaces from input + allSpacesFromInput = false; + break; + } + } else { + // This input row doesn't have content at this column + // Check if we added padding spaces + let hasContentBelow = false; + for (let checkRow = row + 1; checkRow < input.length; checkRow++) { + if (colIndex < input[checkRow].length) { + hasContentBelow = true; + break; + } + } + + if (hasContentBelow) { + // We added a padding space, so this isn't just input spaces + allSpacesFromInput = false; + break; + } + } + } + + if (allSpacesFromInput) { + trailingSpaceRows++; + } else { + break; + } + } else { + break; + } + } + + // If we have trailing space rows that should be combined + if (trailingSpaceRows > 0) { + // Append the trailing spaces to the previous row + const combineStartIndex = result.length - trailingSpaceRows; + const previousRow = result[combineStartIndex - 1]; + let combinedSpaces = ''; + + for (let i = combineStartIndex; i < result.length; i++) { + combinedSpaces += result[i]; + } + + // Replace the previous row with the combined version + result[combineStartIndex - 1] = previousRow + combinedSpaces; + + // Remove the trailing space rows + result.splice(combineStartIndex, trailingSpaceRows); + } + + // Remove any remaining completely empty trailing rows + while (result.length > 0 && result[result.length - 1] === '') { + result.pop(); + } + + return result; +}