题目描述:
将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;
}