洛谷p4933 大师

传送门
第一反应可以当成区间dp做，但是由于等差数列会存在公差为负数的情况，导致需要的空间加倍，故换了一种思路
公差可以通过等差数列的最后两项推出，因此只需要知道相邻两项的高度就可以
以f[i][j]表示等差数列以第i项结尾，倒数第二项是j的方案数
f[i][j]=sum(f[j][k])+1，k<j && h[i]-h[j]==h[j]-h[k]
由于只有两项的时候也是合理方案，因此f[i][j]最小为1
由于只有i一项的时候也是合理方案，但不能被任何一个状态包含，所以只能单独处理

代码

#include <iostream>
#include <vector>
using namespace std;
const int N = 1010, V = 20010, mod = 998244353;
int f[N][N];
int h[N];
//相当于一个map，判断是否有该高度的塔存在，v[i]存放所有高度为i的塔下标
vector<int> v[V];
long long ans;
int main(){
	int n;
	cin>>n;
	for(int i = 0;i < n;i++){
		cin>>h[i];
		v[h[i]].push_back(i);
	}
	for(int i = 0;i < n;i++){
		//单独处理只有一个塔的情况
		ans = (ans + 1) % mod;
		//确定以i为终点，枚举在i前面的所有j
		for(int j = 0;j < i;j++){
			//只有两个塔
			f[i][j] = 1;
			//判断公差
			int len = h[i] - h[j];
			//求出倒数第三个塔的高度，判断满足条件的塔是否存在
			if(h[j] - len <= 20000 && h[j] - len > 0 && v[h[j] - len].size()){
				for(auto t : v[h[j] - len]){
					//有可能有多个高度相等的塔，位置不同时视为两个塔
					if(t < j) f[i][j] = ((long long)f[i][j] + f[j][t]) % mod;
				}
			}
			//每次求出一个状态的合理方案数都应该累加到答案中
			ans = (ans + f[i][j]) % mod;
		}
	}
	cout<<ans<<endl;
	return 0;
}