如何将文本排成相同的高度？

<div class="container">
  <div class="text-block">文本1</div>
  <div class="text-block">文本2</div>
  <div class="text-block">文本3</div>
</div>

<style>
.container {
  display: flex;
  align-items: stretch;
}

.text-block {
  border: 1px solid #000;
  flex: 1;
}
</style>

<div class="text-block">文本1</div>
<div class="text-block">文本2</div>
<div class="text-block">文本3</div>

<script>
var textBlocks = document.getElementsByClassName('text-block');
var maxHeight = 0;

for (var i = 0; i < textBlocks.length; i++) {
  if (textBlocks[i].offsetHeight > maxHeight) {
    maxHeight = textBlocks[i].offsetHeight;
  }
}

for (var i = 0; i < textBlocks.length; i++) {
  textBlocks[i].style.height = maxHeight + 'px';
}
</script>

<table>
  <tr>
    <td>文本1</td>
  </tr>
  <tr>
    <td>文本2</td>
  </tr>
  <tr>
    <td>文本3</td>
  </tr>
</table>