如何使用相同的按钮切换两个元素(无JQuery)

<button id="toggleButton">切换元素</button>
<div id="element1">元素1</div>
<div id="element2">元素2</div>

// 获取按钮和两个元素的引用
var toggleButton = document.getElementById("toggleButton");
var element1 = document.getElementById("element1");
var element2 = document.getElementById("element2");

// 定义一个变量来跟踪当前显示的元素
var currentElement = element1;

// 给按钮添加点击事件监听器
toggleButton.addEventListener("click", function() {
  // 切换元素的显示状态
  if (currentElement === element1) {
    element1.style.display = "none";
    element2.style.display = "block";
    currentElement = element2;
  } else {
    element1.style.display = "block";
    element2.style.display = "none";
    currentElement = element1;
  }
});