我觉得这道题不错!
我们先不要操作几次(因为最后多加一维就行)
然后dp[i][j],表示覆盖完前i个了,最后一段使用j位置的数覆盖的
那么dp[i][j]是由那些转移过来的呢?
1) 首先i到j之间的最大值是a[j],不然i的位置不可能由j的位置的数覆盖。
2) 他可以是任何数覆盖i-1之前的位置转移过来,但是:
假设他前面第一个比他大的数的位置是r,他不能由某个数覆盖r之前的位置转移过来(因为j位置的数不能覆盖之前位置到r这一段)
于是你会发现这其实两个矩阵,维护一下矩阵前缀和就行了
然后注意的一点是,如果只是自己位置覆盖自己位置是不需要操作数的,i==j的时候特殊处理一下就行了。
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <cstring>
using namespace std;
class AlienAndPermutation {
public:
int getNumber(vector<int> , int);
};
long long pmod = 1000000007;
long long dp[205][205][205];
int a[205];
int nt[205], pre[205];
long long sum[205][205][205];
int AlienAndPermutation::getNumber(vector<int> A, int m) {
int i, j, k, n;
memset(dp, 0, sizeof(dp));
n = A.size();
for (i = 0; i < n; ++i) {
a[i + 1] = A[i];
}
a[0] = n + 1;
n++;
for (i = 1; i < n; ++i) {
for (j = i; j >= 0; --j)
if (a[j] > a[i])
break;
pre[i] = j;
for (j = i; j < n; ++j)
if (a[j] > a[i])
break;
nt[i] = j;
}
memset(sum, 0, sizeof(sum));
memset(dp, 0, sizeof(dp));
for (i = 0; i < n; ++i)
dp[0][i][i] = 1;
for (i = 0; i < n; ++i) {
for (j = 0; j < n; ++j) {
if (i == 0 && j == 0)
sum[0][i][j] = dp[0][i][j];
else if (i == 0)
sum[0][i][j] = sum[0][i][j - 1] + dp[0][i][j];
else if (j == 0)
sum[0][i][j] = sum[0][i - 1][j] + dp[0][i][j];
else
sum[0][i][j] = sum[0][i - 1][j] + sum[0][i][j - 1]
+ dp[0][i][j] - sum[0][i - 1][j - 1];
}
}
for (k = 1; k <= m; ++k) {
for (i = 1; i < n; ++i) {
for (j = 1; j < n; ++j) {
if (pre[j] >= i || nt[j] <= i)
dp[k][i][j] = 0;
else if (j == i) {
dp[k][i][j] = sum[k][i - 1][i - 1] - ((i - 2 < 0) ? 0
: sum[k][i - 2][i - 1]);
if (a[i - 1] < a[i]) {
dp[k][i][j] += sum[k - 1][i - 2][i - 1]
- ((pre[j] == 0) ? 0
: sum[k - 1][pre[j] - 1][pre[j]]);
}
dp[k][i][j] %= pmod;
} else {
dp[k][i][j] = sum[k - 1][i - 1][j - 1] - ((pre[j] == 0) ? 0
: sum[k - 1][pre[j] - 1][pre[j]]);
dp[k][i][j] %= pmod;
}
sum[k][i][j] = dp[k][i][j] + sum[k][i - 1][j]
+ sum[k][i][j - 1] - sum[k][i - 1][j - 1];
sum[k][i][j] %= pmod;
}
}
}
long long ans = 0;
for (k = 0; k <= m; ++k)
for (j = 1; j < n; ++j) {
ans += dp[k][n - 1][j];
ans %= pmod;
}
return (ans + pmod) % pmod;
}