力扣1866——恰有 K 根木棍可以看到的排列数目（动态规划）

解题思路

f[i][j]表示前i个木棍看到j个，f[n][k]即为答案；
前i个木棍，i最高一定能被看到，如果他在最后一位，相当于由f[i-1][j-1]转移而来，在原来基础的末尾上添加一个；
如果i不在最后一位，则最后一位一定是被挡掉的，可以有i-1种选择，不妨设选择x；
而前面的i-1个数（0_x-1，x+1i）的组合方案，能看到j个数；
对此将x+1_{i这些数全部-1，令其等于x}i-1，并不影响相对大小关系，看到的数跟原来一样；
例如1,2,4的排列组合方案与1,2,3的个数相同；
所以前面的i-1个数（0_x-1，x+1i）能看到j个数的方案数，就是f[i-1][j]；
所以f[i][j] = f[i-1][j-1] + (i - 1) * f[i-1][j];

f[i]只与f[i-1]有关，可以用两个一维数组滚动更新；

代码

class Solution {
public:
    long long MOD = 1e9 + 7;
    int rearrangeSticks(int n, int k) {
        vector<long long> f(k+1, 0);
        f[0] = 1;
        for(int i = 1; i <= n; i++) {
            vector<long long> g(k+1, 0);
            for(int j = 1; j <= i && j <= k; j++) {
                g[j] = f[j-1] + (i-1) * f[j];
                g[j] %= MOD;
            }
            f = g;
        }
        return f[k];
    }
};