平方等式问题

笔试题：平方等式
题目描述：

公式如：f(13) = 1^2+3^2 = 10;f(23) = 2^2+3^2 = 13;
现要求找出满足范围a–b之间所有，K*f(n) = n的数，输出一共有多少个满足条件的数以及其数本身，其中k为手动输入。

 function lib(k,a,b){
    var sum = 0;
    var key = [];
    for(var i=a;i<b;i++){
        var num = 0;
        var s = String(i).split(''); //将每一个数转化为数组形式
        var len = s.length;
        for(var j=0;j<len;j++){
            num = num+Math.pow(s[j],2); //求每一个数的平方和
        }
        if(k*num == i){
            sum++;
            key.push(i); //将符合条件的数放进数组中
        }
    }
    var ty = [sum,key];
    return ty;
}
var k=parseInt(readline()),         
    a=parseInt(readline()),
    b=parseInt(readline());
var res = lib(k,a,b);
console.log(res);