fork(3) download
  1. import java.util.Arrays;
  2. import java.util.stream.IntStream;
  3. import java.util.stream.Stream;
  4.  
  5. class Main {
  6. public static void main(String[] args) throws Exception {
  7. int[][] matrix = getTimeMatrix();
  8.  
  9. int[][] result = Stream.of(matrix)
  10. .map(a -> IntStream.of(a).filter(i -> i != 0).count())
  11. .map(c -> IntStream.range(0, c.intValue()).toArray())
  12. .toArray(int[][]::new);
  13.  
  14. System.out.println(Arrays.deepToString(result));
  15. }
  16.  
  17. static int[][] getTimeMatrix() {
  18. return new int[][] { { 10, 20, 0, 1 }, { 1, 0, 0, 50 }, { 1, 1, 1, 1 } };
  19. }
  20. }
Success #stdin #stdout 0.14s 2184192KB
stdin
Standard input is empty
stdout
[[0, 1, 2], [0, 1], [0, 1, 2, 3]]