การนับตัวเลขเฉพาะที่ใช้ในกำลังสองของตัวเลขใน JavaScript

ปัญหา

เราต้องเขียนฟังก์ชัน JavaScript ที่รับจำนวนเต็ม n (n >= 0) และหลักฐาน d (0 <= d <= 9) ฟังก์ชันควรยกกำลังสองตัวเลข k ทั้งหมด (0 <= k <= n) ระหว่าง 0 ถึง n และนับจำนวนหลักฐาน d ที่ใช้ในการเขียน k**2 ทั้งหมด

ตัวอย่าง

โค้ดตัวอย่างดังนี้:

const n = 25;
const d = 1;
const countDigits = (n, d) => {
  let k = 0, count = 0;
  d = d.toString();
  while (k <= n) {
    let a = 0;
    let s = (k*k).toString();
    for(let i = 0; i < s.length; i++)
      if(s[i] == d)
        a++;
    if (a > 0) {
      count += a;
    };
    k++;
  };
  return count;
};
console.log(countDigits(n, d));

ผลลัพธ์

11