网易内推笔试题

题目点我

小易喜欢的数列：
小易非常喜欢拥有以下性质的数列:
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

输出例子1:
3

就是dp，但是普通的做法会超时。
超时的做法：用f(k, n)表示数字k打头， 长度为n的数列的个数，那么有

$$f(k, n) = \sum{f(j, n - 1)}, j \geq k \ ||\ k\ mod\ j\neq \ 0$$
然后三重循环自底向上求解。
上面做法超时的原因在于对于f(k, n)来说，长度为n - 1的情况大部分都是要加上去的，因为j % k == 0的情况毕竟是稀疏的，所以对不同的k来说，用往f(k, n)上累加合法情况的做法重复计算很多。

既然不合法的状况是稀疏的，我们可以对所有长度为n的序列统一计算sum(f(j, n - 1))，然后根据不同的k从中剔除掉不符合的f(j, n - 1)即可。然而用k打头的状态定义循环不太好写，因为要固定k，循环j找到所有的k % j == 0。
如果用另一种状态表示：g(k, n)表示结尾数字为k，长度为n的数列个数，那么状态转移方程就是:

$$g(k, n) = \sum_{1 \leq j \leq k}{g(j, n - 1)} - \sum_{i = multiple\ k }{g(i, n - 1)}$$
其中对g(j, n - 1)求和的部分，即 $\sum_{1 \leq j \leq k}{g(j, n - 1)}$，对所有长度为n的数列都可以复用，节省计算量。给定结尾数字j之后，不符合的情况就是k前面的数字是2j, 3j, 4j, ….所有不超过k的j的倍数。

#include <iostream>
#include <vector>
#include <deque>
#include <string>
#include <algorithm>

using namespace std;

const int mod = 1000000007;

int main() {
    int n, k;
    cin >> n >> k;
    vector<vector<int>> dp(k + 2, vector<int>(n + 3, 0));
    for (int i = 1; i <= k; ++i) {
        dp[i][1] = 1;
    }
    for (int len = 2; len <= n; ++len) {
        int total = 0;
        for (int end_num = 1; end_num <= k; ++end_num) {
            total += dp[end_num][len - 1];
            total %= mod;
        }
        for (int current_num = 1; current_num <= k; ++current_num) {
            int total2 = 0;
            for (int prev_num = 2 * current_num; prev_num <= k; prev_num += current_num) {
                total2 += dp[prev_num][len - 1];
                total2 %= mod;
            }
            dp[current_num][len] = (total - total2 + mod) % mod;
        }
    }
    unsigned long res = 0;
    for (int i = 1; i <= k; ++i) {
        res += dp[i][n];
    }
    res = res % mod;
    cout << res << endl;
    return 0;
}