2.8-2求解幸运数问题

题目描述：

小明同学学习了不同的进制之后，拿起了一些数字做起了游戏。小明同学知道，在日常生活中我们最常用的是十进制数，而在计算机中，二进制数也很常用。现在对于一个数字x，小明同学定义出了两个函数f(x)和g(x)。 f(x)表示把x这个数用十进制写出后各个数位上的数字之和。如f(123)=1+2+3=6。 g(x)表示把x这个数用二进制写出后各个数位上的数字之和。如123的二进制表示为1111011，那么，g(123)=1+1+1+1+0+1+1=6。 小明同学发现对于一些正整数x满足f(x)=g(x)，他把这种数称为幸运数，现在他想知道，大于0且小于等于n的幸运数有多少个？

举例：

输入
21

输出
3

#include <bits/stdc++.h>

using namespace std;

int getSum(int i,int k) {
	int sum = 0;
	while(i > 0) {
		sum += i % k;
		i /= k;
	}
	return sum;
}


int main() {
	int n;
	cin >> n;
	int ans;
	for(int i = 1; i <= n; i++) {
		int g = getSum(i,10);
		int f = getSum(i,2);
		ans += f == g ? 1 : 0;
	}
	
	cout << ans << endl;
	
	return 0;
}