数列-找规律-快速幂-python题解

数列

给定一个正整数k(3≤k≤15),把所有k的方幂及所有有限个互不相等的k的方幂之和构成一个递增的序列，例如，当k=3时，这个序列是：
　　1，3，4，9，10，12，13，…
　　（该序列实际上就是：30，31，30+31，32，30+32，31+32，30+31+32，…）
　　请你求出这个序列的第N项的值（用10进制数表示）。
　　例如，对于k=3，N=100，正确答案应该是981。

【输入形式】
只有1行，为2个正整数，用一个空格隔开：
　　k N
　　（k、N的含义与上述的问题描述一致，且3≤k≤15，10≤N≤1000）。

【输出形式】
计算结果，是一个正整数（在所有的测试数据中，结果均不超过2.1*109）。（整数前不要有空格和其他符号）。
【样例输入】

3 100

【样例输出】

981

提供一个算法,其实由于是将有限个互不相等的k,所以我们这里考察第n项的时候,能取到的最大的k的幂次,不妨设次数是p,那么与之相关的很显然应该有2^p个,于是就想到了其实an与k的次数是和n的二进制有关的.
假设n=∑bk2^{k,bk=1或0.于是我们有an=∑(bk*k}(k-1)).
比如说k=3,第7项,因为7对应的二进制数是111(2),所以a7=13^2+1*31+13^0=13.
若为第11项,11对应的二进制是1011(2),所以a11=13^3+0*32+1*3^1+1*30=31.

k,n=map(int,input().split())
s=list(bin(n)[2:])
def q(x,y):
    res=1
    while y>0:
        if y&1:
            res*=x
        x*=x
        y=y>>1
    return res
ans=0
for i in range(len(s)):
    x=int(s[i])
    y=len(s)-1-i
    ans+=x*q(k,y)
print(ans)