校招真题练习035 最少立方数之和(小米)

最少立方数之和

题目描述
给出一个数字N（0<N<1000000）,将N写成立方数和的形式，求出需要的最少立方数个数。
例如N=17，1+8+8 = 17，最少需要3个立方数，则输出3。
N= 28,1+1+1+1+8+8+8=28, 需要7个立方数，1+27=28,需要2个立方数，所以最少立方数为2，则输出2。

输入描述:
一个数字N（0<N<1000000）

输出描述:
最少立方数个数

def main():
    n = int(input())
    power = set()
    base = 1
    while base**3 <= n:
        curnum = base**3
        if curnum == n:
            print(1)
            return
        power.add(curnum)
        base += 1
    level = 1
    target = {n}
    while target:
        cur = set()
        for i in target:
            for e in power:
                if i-e in power:
                    print(level+1)
                    return
                if i-e > 0:
                    cur.add(i-e)
        target = cur
        level += 1

if __name__ == '__main__':
    main()

算法思路：BFS