小于等于n的幸运数有多少个

#include <iostream>
using namespace std;

int f(int x)
{
    int sum = 0;
    while (x > 0)
    {
        sum += x % 10;
        x = x / 10;
    }
    return sum;
}

int g(int x)
{
    int sum = 0;
    while (x > 0)
    {
        sum += x % 2;
        x = x / 2;
    }
    return sum;
}

int count_lucky_number(int n)          //计算小于等于n的幸运数个数
{
    int count, x;
    count = 0;
    for (x = 1; x <= n; x++)
    {
        if (f(x) == g(x))
            count++;
    }
    return count;
}

int main(void)
{
    int n;
    cin >> n;
    cout << count_lucky_number(n) << endl;
    return 0;
}