如何使用 JavaScript 實作圖片瀑布流佈局？

<div id="waterfall-container">
  <div class="column">
    <img src="image1.jpg">
  </div>
  <div class="column">
    <img src="image2.jpg">
  </div>
  ...
</div>

#waterfall-container {
  width: 1000px;
  column-count: 4;
  column-gap: 20px;
}

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

window.onload = function() {
  var container = document.getElementById("waterfall-container");
  var columns = container.getElementsByClassName("column");

  function calculateColumnHeight() {
    var containerWidth = container.offsetWidth;
    var columnWidth = containerWidth / columns.length;

    for (var i = 0; i < columns.length; i++) {
      var column = columns[i];
      var images = column.getElementsByTagName("img");
      var totalImageHeight = 0;

      for(var j = 0; j < images.length; j++) {
        var image = images[j];
        var imageWidth = image.offsetWidth;
        var imageHeight = image.offsetHeight;
        var imageRatio = imageWidth / imageHeight;
        var adjustedImageHeight = columnWidth / imageRatio;

        totalImageHeight += adjustedImageHeight;
      }

      column.style.height = totalImageHeight + "px";
    }
  }

  calculateColumnHeight();
  window.addEventListener("resize", calculateColumnHeight);
}

function addNewImage(imageUrl) {
  var container = document.getElementById("waterfall-container");
  var columns = container.getElementsByClassName("column");
  
  var minHeightColumn = columns[0];
  for (var i = 1; i < columns.length; i++) {
    if (columns[i].offsetHeight < minHeightColumn.offsetHeight) {
      minHeightColumn = columns[i];
    }
  }
  
  var newImage = document.createElement("img");
  newImage.src = imageUrl;
  minHeightColumn.appendChild(newImage);
  
  calculateColumnHeight();
}

addNewImage("image3.jpg");