统计数字问题

问题描述：
一本书的页码从自然数1开始顺序编码直到自然数n。书的页码按照通常的习惯编排，每个页码都不含多余的前导数字0。例如第6页用6表示，而不是06或006。数字统
计问题要求对给定书的总页码n，计算出书的全部页码中分别用到多少次数字0,1,2,3,…,9。
编程任务：
给定表示书的总页码的十进制整数n(1<<n<<109)。编程计算书的全部页码中分别用到多少次数字0,1,2,3,…,9。
数据输入：
输入数据由文件名input.txt的文本文件提供。每个文件只有1行，给出表示书的总页码的整数n。

结果输出：

程序运行结束时，将计算结果输出到文件output.txt中。输出文件共有10行，在第k行输出页码中用到数字k-1的次数，k=1,2,3,…,10。

Python代码如下：

def read_file(filename):
    f=open(filename)
    n=f.readline()
    return int(n)

def write_file(filename,dictn):
    f=open(filename,'w')
    for i in xrange(len(dictn)):
        f.write(str(dictn[i])+'\n')
    f.close()
    
    
    
def countN(n):
	dictn={}.fromkeys([0,1,2,3,4,5,6,7,8,9],0)
	for i in xrange(1,n+1):
		tmp=i
		while tmp>=10:
			dictn[tmp%10]+=1
			tmp/=10
		dictn[tmp]+=1
	return dictn
        
n=read_file("input.txt")
dic=countN(n)
write_file("output.txt",dic)