wa5个 数组没有开小(吧)

#include <bits/stdc++.h>
using namespace std;
const int MOD = 998244353;
int n,h[4005],dp[4005][80010],ans = 0;
int main(){
	cin >> n;
	for(int i = 1;i <= n; i++)
		cin >> h[i];
	for(int i = 1;i <= n; i++){
		for(int j = 1;j < i; j++){
			dp[i][h[i]-h[j]+40000] += dp[j][h[i]-h[j]+40000]+1;
			dp[i][h[i]-h[j]+40000] %= MOD;
			ans += (dp[j][h[i]-h[j]+40000]+1)%MOD;
		}
	}
	cout << ans+n;
	return 0;
}