【洛谷4933】大师（DP）

题目：

洛谷4933

分析：

（自己瞎yy的DP方程竟然1A了，写篇博客庆祝一下）

（以及特斯拉电塔是向Red Alert致敬吗233）

这里只讨论公差不小于$0$的情况，小于$0$的情况进行复读机即可（注意不要重复计算公差为$0$的情况）。

用$dp[i][j]$表示结尾为第$i$个数，公差为$j$的长度不小于$2$的非降等差数列的方案数（单独$1$个数的情况公差不确定不好处理，最后给答案加上$n$就行了）。

那么对于$i$，枚举所有$j(j<i$且$h[j]\le 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 
#include 
#include 
#include 
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();
}