题目:
分析:
(自己瞎yy的DP方程竟然1A了,写篇博客庆祝一下)
(以及特斯拉电塔是向Red Alert致敬吗233)
这里只讨论公差不小于\(0\)的情况,小于\(0\)的情况进行复读机即可(注意不要重复计算公差为\(0\)的情况)。
用\(dp[i][j]\)表示结尾为第\(i\)个数,公差为\(j\)的长度不小于\(2\)的非降等差数列的方案数(单独\(1\)个数的情况公差不确定不好处理,最后给答案加上\(n\)就行了)。
那么对于\(i\),枚举所有\(j(j<i\)且\(h[j]\leq h[i])\),则有
(加\(1\)是加上了新的长度为\(2\)的数列\(\{h[j],h[i]\}\))
\[dp[i][h[i]-h[j]]=\sum_j (dp[j][h[i]-h[j]]+1)
\]
\]
总的来说还是一道比较基础的DP题。时间复杂度\(O(nV)\)
代码:
先念诗:
瓜之嘴
作者:YYC神犇
好山配好水,
好瓜配好嘴。
不膜神仙瓜,
写题一定萎。
(1A多亏模数叫Jumpmelon)
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
using namespace std;
namespace zyt
{
typedef long long ll;
const int N = 1010, V = 2e4 + 10, Jumpmelon = 998244353;
int h[N], n, dp[N][V], ans;
int work()
{
cin >> n;
for (int i = 1; i <= n; i++)
cin >> h[i];
for (int i = 1; i <= n; i++)//upper
{
for (int j = 1; j < i; j++)
if (h[j] <= h[i])
dp[i][h[i] - h[j]] = (dp[i][h[i] - h[j]] + dp[j][h[i] - h[j]] + 1) % Jumpmelon;
}
for (int i = 1; i <= n; i++)
for (int j = 0; j < V; j++)
ans = (ans + dp[i][j]) % Jumpmelon;
memset(dp, 0, sizeof(dp));
for (int i = 1; i <= n; i++)//lower
{
for (int j = 1; j < i; j++)
if (h[i] < h[j])
dp[i][h[j] - h[i]] = (dp[i][h[j] - h[i]] + dp[j][h[j] - h[i]] + 1) % Jumpmelon;
}
for (int i = 1; i <= n; i++)
for (int j = 0; j < V; j++)
ans = (ans + dp[i][j]) % Jumpmelon;
cout << (ans + n) % Jumpmelon;
return 0;
}
}
int main()
{
return zyt::work();
}