洛谷 P4933 大师

Description

建筑大师最近在跟着数学大师 ljt12138 学数学，今天他学了等差数列，ljt12138 决定给他留一道练习题。

ljt12138 首先建了 $n$ 个特斯拉电磁塔，这些电塔排成一排，从左到右依次标号为 $1$ 到 $n$ ，第 $i$ 个电塔的高度为 $h[i]$ 。

建筑大师需要从中选出一些电塔，然后这些电塔就会缩到地下去。这时候，如果留在地上的电塔的高度，从左向右构成了一个等差数列，那么这个选择方案就会被认为是美观的。

建筑大师需要求出，一共有多少种美观的选择方案，答案模 $998244353$ 。

注意，如果地上只留了一个或者两个电塔，那么这种方案也是美观的。地上没有电塔的方案被认为是不美观的。

同时也要注意，等差数列的公差也可以为负数。

Input

第一行一个正整数 $n$ 。

第二行 $n$ 个非负整数，第 $i$ 个整数是第 $i$ 个电塔的高度 $h[i]$ 。

Output

输出一个整数，表示美观的方案数模 $998244353$ 的值。

Solution

1.考虑dp做法

令dp[i][j]为最后为第i座电塔，公差为j时的方案数。

显然，答案为全部dp[i][j]的结果之和。

注意：数组下标没有负数，公差有负数，根据数据范围，公差在$[-20000,20000]$区间，令公差加上一个$delta=20000$即可（dp开到1000*40000）。

2.dp过程

枚举每一座电塔i，先将ans++（自身方案数为1）

在此基础上，枚举i的前面的所有塔j，把相同公差的dp值加到第i座塔的相同dp值上，再加1，即：

dp[i][h[i]-h[j]+delta] += dp[j][h[i]-h[j]+delta]+1;

加一是因为，前j个塔的方案中是没有考虑第i个塔的，也就是相当于第i座塔是缩下去的，现在考虑到了第i个塔，第i个塔可以不缩下去了，所以要在原有方案数上加一。

Code

/*************************************
 * @problem:      problem title.
 * @author:       iamshroud.
 * @time:         2021-??-??.
*************************************/
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <list>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>
// #include <unordered_map>

typedef long long ll;

#define itn int //ovo
#define MOD 998244353

using namespace std;

inline ll read()
{
	ll x=0,f=1;char ch=getchar();
	while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
	while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
	return x*f;
}


ll n;
ll h[1005];
// 前i个结尾的，公差为j的方案数
ll dp[1005][40005];
ll ans;

void solve()
{
    n = read();
    for(int i=1; i<=n; i++)
    {
        h[i] = read();
    }

    ll delta = 20000; 
    ans = n;// 加上单独一个的情况
    for(int i=1; i<=n; i++)
    {
        // 枚举前面的塔
        for(int j=i-1; j>=1; j--)
        {
            // 将前j个的方案加1加到前i个中
            // 前j个的方案是第i个电塔缩下去的
            // 现在可以把电塔升起来，方案加1
            dp[i][h[i]-h[j]+delta] += dp[j][h[i]-h[j]+delta]+1;
            dp[i][h[i]-h[j]+delta] %= MOD;

            ans += dp[j][h[i]-h[j]+delta]+1;
            ans %= MOD;
        }
    }

    cout << ans;
    
}

int main(void)
{
    
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    // come on
    solve();

    


    return 0;
}