【洛谷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]\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();
}