align children with different heights

Question

Is it possible to create a flexbox layout with wrapping children and aligning them to the top?

.container {
  display: flex;
  flex-wrap: wrap;
  justify-content: space-between;
}

.container .element {
  flex: 0 0 auto;
}

creates this:

but my plan is to achieve this:

Answer 1

.container {
  border: 1px solid black;
  padding: 10px;
  height: 250px;
  
  display: flex;
  flex-wrap: wrap;
  flex-direction: column;
  align-content: flex-start;
  
  overflow: hidden;
  resize: both;
}

.container div {
  border: 5px solid #AAA;
  background-color: #DDD;
  margin: 3px;
  height: 100px;
  width: 50px;
}

.container div:nth-child(3n) {
  height: 40px;
}

<div>
  <div class="container">
    <div></div><div></div><div></div>
    <div></div><div></div><div></div>
    <div></div><div></div><div></div>
  </div>
</div>
Some Text