如何用JS更改文档中所有DIV元素的CSS CLASS？

<html>
<head>
<style type="text/css">
.div1{ background-color: red; width:40; height:20}
.div2{ background-color: blue; width:40; height:20}
</style>
<script language="JavaScript">
<!--
function setStyle(flag){
var divs = document.getElementsByTagName("div");

if(flag==1)
for(var i=0;i<divs.length;i++){
divs[i].setAttribute("className","div1");
}
else
for(var i=0;i<divs.length;i++){
divs[i].setAttribute("className","div2");
}
}
//-->
</script>
</head>
<body>
<div>1234</div>
<div>1234</div>
<div>1234</div>
<div>1234</div>
<div>1234</div>
<script language="JavaScript">
<!--
setStyle(1);
//setStyle(2);
//-->
</script>
</body>
</html>