Посчитать количество чисел состоящих из одинаковых цифр

#include <iostream>
using namespace std;
 
long getLeftDigit(long x) {
    while (x / 10 != 0) {
        x /= 10;
    }
 
    return (x % 10);
}
 
long getLength(long x) {
    long len = 0;
    while (x != 0) {
        x /= 10;
        len++;
    }
 
    return len;
}
 
long getPerfectNumber(long len, long d) {
    long x = 0;
    for (long i = 0; i < len; i++) {
        x = x * 10 + d;
    }
 
    return x;
}
 
int main() {
    long n, k;
    cout << "Enter numbers n & k: ";
    cin >> n >> k;
 
    long d_n = getLeftDigit(n);
    long d_k = getLeftDigit(k);
    long len_n = getLength(n);
    long len_k = getLength(k);
 
    long count = 0;
    if (len_n == len_k) {
        count = (d_k - d_n - 1) + (n <= getPerfectNumber(len_n, d_n)) + (k >= getPerfectNumber(len_k, d_k));
    }
    else {
        // Левый интервал.
        count += (9 - d_n) + (n <= getPerfectNumber(len_n, d_n));
 
        // Средний интервал.
        count += 9 * (len_k - len_n - 1);
 
        // Правый интервал.
        count += (d_k - 1) + (k >= getPerfectNumber(len_k, d_k));
    }
 
    cout << "Count: " << count << endl;
 
    return 0;
}