洛谷刷题C++语言 | P4933 大师

​学习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