Заполнить таблицу n * n числами от 1 до n^2 и закрасить все квадраты

var tr=document.getElementsByTagName('tr');
var num=tr.length;
var set=1
for (i = 0; i < num; i++){
    for (k = 0; k < num; k++){
        var td=tr[i].getElementsByTagName('td');
        td[k].textContent=set;
        set+=1;
    }
}

for (i = 0; i < num; i++){
    for (k = 0; k < num; k++){
        var td=tr[i].getElementsByTagName('td');
        td[k].style.backgroundColor="red"; // эту строку добавил
        td[k].textContent=set;
        set+=1;
    }
}

function table(){
 var n = document.form1.peremen.value;
var x = new Array();
 
 
    var tab = '<table  border="2"  bordercolor="#000000" cellspacing="5" cellpadding="5" >';
    for (var i = 0; i < n; i++)
    {
        tab += '<tr>';
        for (var j =0; j < n; j++)
        { 
x=j+1 + (i) * n;
 
tab +=  '<td>'+ x + '</td>';
 
 
        }
        tab += '</tr>';
    }
    tab += '</table>';
      var elem = document.getElementById("tablicta");
    elem.innerHTML =tab;
}

if (/1|4|9/.test(set)) td[k].style.backgroundColor="red";

for (var i=0; i<n; i++)
    { tab += '<tr>';
      for (var j=0; j<n; j++)
          { var x = (j+1+i*n);
 
            if (Math.sqrt(x)%1==0)  //Проверка числа на "квадратность"
              { tab += '<td style="background-color: LightGray;">'+ x + '</td>'; }
            else
              { tab += '<td>'+ x + '</td>'; }
          }
      tab += '</tr>';
    }

var tb = document.getElementById("tablicta");
    var tr = tb.getElementsByTagName("tr");
    var num=tr.length;
    var set=1;
     k=1;
      
     for (i = 0;i < num; i++ ) 
        {        
        var td=tr[i].getElementsByTagName('td');
        for (j = 0; j < num; j++ )
             {        
                   if (set==k*k)
                   { td[j].style.backgroundColor="red";
                     td[j].textContent=set;
                   }
                 else 
                 td[j].style.backgroundColor="#FFFFCC"
                      
           set=set+1;  } 
           k=k+1; 
   }

function testNum(n) 
{ 
    var i, j, b=0;
    
    for (i=1; i*i<n; i++)
    { for (j=1; j*j<n; j++)
       { if (i*i+j*j==n) b=1; }
    }
 
    return b;
}