Problem Description
Example:
Input:
1 2 3
4 5 6
7 8 9
Return the following:
[
[1],
[2, 4],
[3, 5, 7],
[6, 8],
[9]
]
Input:
1 2
3 4
Return the following:
[
[1],
[2, 3],
[4]
]
NOTE: You only need to implement the given function. Do not read input, instead use the arguments to the function. Do not print the output, instead return values as specified. Still have a question? Checkout Sample Codes for more details.