招聘编程题0x04

#! python3
#-*- coding:utf-8 -*-

import sys

'''
题目描述
小明同学学习了不同的进制之后，拿起了一些数字做起了游戏。
小明同学知道，在日常生活中我们最常用的是十进制数，而在计算机中，二进制数也很常用。
现在对于一个数字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)，
他把这种数称为幸运数，现在他想知道，小于等于n的幸运数有多少个？
输入描述:
每组数据输入一个数n(n<=100000)
输出描述:
每组数据输出一行，小于等于n的幸运数个数。
示例1
输入
21
'''

def f(X):
    "将数字转字符串,计算每位和"
    X=str(X)
    return sum(int(i) for i in X)

def g(X):
    "利用bin()转化为二进制,数字转字符串,计算每位和"
    X=str(bin(X))[2:]
    return sum(int(i) for i in X)

def luckey_num(n):
    "doc"
    Sum=0
    n+=1
    for i in range(1,n):
        if f(i)==g(i):
            Sum+=1
    return Sum 

if __name__ == '__main__':
    "doc"
    for line in sys.stdin:
        a=int(line.strip())
        print(luckey_num(a))