2018网易校招内推-小易喜欢的数列

题目描述：

小易非常喜欢拥有以下性质的数列:
1、数列的长度为n
2、数列中的每个数都在1到k之间(包括1和k)
3、对于位置相邻的两个数A和B(A在B前),都满足(A <= B)或(A mod B != 0)(满足其一即可)

例如,当n = 4, k = 7
那么{1,7,7,2},它的长度是4,所有数字也在1到7范围内,并且满足第三条性质,所以小易是喜欢这个数列的
但是小易不喜欢{4,4,4,2}这个数列。小易给出n和k,希望你能帮他求出有多少个是他会喜欢的数列。

输入描述:

输入包括两个整数n和k(1 ≤ n ≤ 10, 1 ≤ k ≤ 10^5)

输出描述:

输出一个整数,即满足要求的数列个数,因为答案可能很大,输出对1,000,000,007取模的结果。

示例1
输入

2 2

输出

3

解题分析：

设置一个二维状态数组state[[]]，这个状态数组第一维为数列的每个数的情况，例如state=[[0, 1, 1, 1, 1], [0, 4, 3, 3, 2], [0, 12, 8, 8, 5]]，表示state[0]、state[1]、state[2] 第一个数、第二个数、第三个数的状态情况。如前面的数取1，后面可以是（1，2，3，4），如果取2，后面可以是（2，3，4），如果取3，后面可以是（2，3，4）如果取4，（3，4），这样即为我们上述的state的第二个子序列的状态情况，[4,3,3,2]，下面的做法原理一样。

实现代码如下：

def func():
    nums = [int(item) for item in raw_input().split()]
    n,k1 = nums[0],nums[1]
    state = [[0 for i in range(k1+1)] for j in range(n)]
    for j in range(n):
        for i in range(1,k1+1):
            if j == 0:
                state[j][i] = 1
            else:

                ## i为头，看k后面可以跟多少个数
                for k in range(1,k1+1):
                    if k >= i or i%k!=0:
                        state[j][i] += state[j-1][k]
                state[j][i] %= 1000000007

    return sum(state[n-1])%1000000007
if __name__ == '__main__':
    state=func() 
    print state