大师

合集 - 洛谷刷题(22)

1.找筷子（异或运算）01-012.路标设置（二分答案）01-023.单词背诵（滑动窗口）01-034.修复公路（并查集）01-045.细胞分裂（质因数分解）02-246.编辑距离（二维dp）02-257.没有上司的舞会（树形dp）02-288.负进制转换03-029.最大食物链计数03-0410.摆花03-0411.线段03-0412.打鼹鼠03-0613.琪露诺03-06

14.大师03-06

15.时间复杂度03-0816.营业额统计03-0917.选数03-0918.删数问题03-1219.最大路径异或和03-1320.八数码（BFS版本）03-1421.求和03-1522.银河英雄传说03-16

收起

题目链接：https://www.luogu.com.cn/problem/P4933

题意：

给定数组，每次可以任选数组中的元素消除，求能构成等差数列的方案数

思路：

记 dp[i,j] 为以i作结尾，公差为j的等差数列数量

枚举i之前的那些数k

j=a[i]-a[k]

dp[i,j] <-- dp[k,j] +1 (因为k，i可以构成有两个元素的等差数列）

因为公差可以相同并且通过不同位置转移的方案数需要叠加，所以dp[i,j]需要+=

每一个以i结尾的还有单独它一个数的情况

注意公差可以为负，所以需要将其离散

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define endl "\n"
#define int long long
#define fi first
#define se second
//#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef __int128 lll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=1e5+5;
const int mod=998244353;
const int N=2e4+5;
const int p=2e4;
int ans;
int cal(int cnt){
	return cnt*(cnt+1)/2;
}
void solve(){
	int n;cin>>n;
	vector<int>a(n+1,0);
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	vector<vector<int>>dp(n+1,vector<int>(2*N,0));
	
	for(int i=1;i<=n;i++){
		
		for(int j=1;j<=i-1;j++){
			int dif=a[i]-a[j]+p;			
			dp[i][dif]=(dp[i][dif]%mod+(dp[j][dif]+1)%mod)%mod;	
			ans=(ans%mod+(dp[j][dif]+1)%mod)%mod;
		}
		ans=(ans+1)%mod;
	}
	cout<<ans<<endl;
}
 
signed main()
{
	ios::sync_with_stdio(false),cin.tie(0);
	int T=1;
	
	while(T--){
	solve();
	}
	
	return 0;
}