Đếm chữ số cụ thể trong bình phương các số bằng JavaScript

Đề bài

Viết hàm nhận số nguyên n (n ≥ 0) và chữ số d (0 ≤ d ≤ 9). Bình phương mọi số k (0 ≤ k ≤ n), đếm tổng số lần chữ số d xuất hiện trong các bình phương đó.

Ví dụ

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

Kết quả

11