20210808 拼多多笔试第四题

题目描述：

将N道笔试题进行排序，只要相邻的题目中，前一道题的难度不超过后一道题难度+M，就认为是满足序列。问 满足的序列有多少种

输入：

第一行 N，M (N是题目的个数)

第二行 N个数，代表N道题的难度

用例一

输入

4 0

1 2 2 3

输出

2

用例二

输入

3 1

1 2 3

输出

4

方法一、

dfs列举N！种排列方式，每种排列判断是否符合条件

只能过30%

方法二、

举例：对于 1 4 7 3

首先进行排序得到 1 3 4 7，整个问题抽象为一个dp问题，dp[i]表示0~i长度的序列有多少种排列方式，对于7来说，它可以放到3前面，4前面，和当前位置。

因此遍历排序后的数组，判断每个值能插入到前面的个数

#include <iostream>
#include <vector>
#include <cstdio>
#include <algorithm>

using namespace std;

int main() {
    freopen("in.txt", "r", stdin);
    int N, M;
    cin >> N >> M;
    int ori = N;
    int tmp;
    vector<int> prob;
    while(ori--) {
        cin >> tmp;
        prob.push_back(tmp);
    }
    sort(prob.begin(), prob.end());
    vector<int> dp(N, 0);
    dp[0] = 1;
    for(int i = 1; i < N; ++i) {
        int count = 0;
        for(int j = i - 1; j >= 0 && prob[j] + M >= prob[i]; j--)
            count++;
        dp[i] = dp[i-1] * (count + 1);
    } 
    cout << dp.back() << endl;
}