像pinterest那样让图片适合div的好方法是什么？

<div class="container">
  <div class="column">
    <img src="image1.jpg" alt="Image 1">
  </div>
  <div class="column">
    <img src="image2.jpg" alt="Image 2">
  </div>
  <div class="column">
    <img src="image3.jpg" alt="Image 3">
  </div>
  <!-- 更多图片元素... -->
</div>

.container {
  column-count: 3; /* 列数 */
  column-gap: 10px; /* 列间距 */
}

.column {
  display: inline-block;
  width: 100%;
}

img {
  width: 100%;
  height: auto;
}

window.onload = function() {
  var columns = document.getElementsByClassName('column');
  var minHeight = 0;

  for (var i = 0; i < columns.length; i++) {
    var columnHeight = columns[i].offsetHeight;
    if (minHeight === 0 || columnHeight < minHeight) {
      minHeight = columnHeight;
    }
  }

  for (var i = 0; i < columns.length; i++) {
    columns[i].style.height = minHeight + 'px';
  }
};