学习C++从娃娃抓起!记录下洛谷C++学习和备考过程中的题目,记录每一个瞬间。
附上汇总贴:洛谷刷题C++语言 | 汇总
【题目描述】
ljt12138 首先建了 n 个特斯拉电磁塔,这些电塔排成一排,从左到右依次标号为 1 到 n,第 i 个电塔的高度为 h[i]。
建筑大师需要从中选出一些电塔,然后这些电塔就会缩到地下去。这时候,如果留在地上的电塔的高度,从左向右构成了一个等差数列,那么这个选择方案就会被认为是美观的。
建筑大师需要求出,一共有多少种美观的选择方案,答案模 998244353。
注意,如果地上只留了一个或者两个电塔,那么这种方案也是美观的。地上没有电塔的方案被认为是不美观的。
同时也要注意,等差数列的公差也可以为负数。
【输入】
第一行一个正整数 n。
第二行 n 个非负整数,第 i 个整数是第 i 个电塔的高度 h[i]。
【输出】
输出一个整数,表示美观的方案数模 998244353 的值。
【输入样例】
8
13 14 6 20 27 34 34 41
【输出样例】
50
【代码详解】
#include <bits/stdc++.h>
using namespace std;
const int N = 1005;
int n, ans, a[N], f[N][N];
int main()
{
cin >> n;
for (int i=1; i<=n; i++) cin >> a[i];
ans += n; // 1个数也可以构成一个美观的方案
for (int i=1; i<n; i++) {
for (int j=i+1; j<=n; j++) {
f[i][j] = 1;
int x = a[j]-a[i];
for (int k=1; k<i; k++) {
if (x==a[i]-a[k]) {
f[i][j] = (f[i][j] + f[k][i])%998244353;
}
}
ans = (ans+f[i][j])%998244353;
}
}
cout << ans;
return 0;
}
【运行结果】
8
13 14 6 20 27 34 34 41
50