How to make a similar grid on flexbox?

Question

css

Hello!
The task is to correctly arrange the elements depending on the height.
Here is an example:

It is necessary if one element has a large height, then the others lined up to the right / left of it.
The picture shows the desired behavior, the styles I wrote are what I have now.
How can such behavior be implemented?
Thanks