agc041d

agc041d Problem Scores

题目链接

题目大意

给出数列长度 $N$ ($N\le 5000$)与质数 $M$ ($M\le 10^9$) 输出满足以下三个条件的数列的数目
$$\forall i\in [1,n),a_i\le a_{i+1}$$

$$\forall i\in [1,n],1\le a_i\le n$$

$$\forall k\in [1,n),\sum _{i=1}^{k+1}{a}_i>\sum _{i=n-k+1}^n{a}_i$$

思路

条件一可以简述为数列单调不降

条件二可以通过条件一化简为$a_1\ge 1$ 与$a_n\le n$

条件三较为复杂,对其进行分析

当前$k+1$ 个数与后$k$ 个数重叠时,考虑去掉重叠的数,可以转化为不重叠的情况

因此条件三只需要考虑$\forall k\in [1,\lfloor \frac{n-1}{2}\rfloor ]$ 的情况即可

观察该区间内的不等式,发现若对于$k=x$ 不等式成立,则对于$k=x-1$ 不等式依旧成立

因为$k$ 从$x$ 到$x-1$ ,相当于不等号左边减掉一个较小的数,右边减掉一个较大的数,所以依旧成立

因此,如果对于$k=\lfloor \frac{n-1}{2}\rfloor$ 有不等式成立,则对于$\forall k\in [1,\lfloor \frac{n-1}{2}\rfloor ]$ ,不等式均成立

现给出条件四
$$k=\lfloor \frac{n-1}{2}\rfloor ,\sum _{i=1}^{k+1}{a}_i>\sum _{i=n-k+1}^n{a}_i$$
综上,条件四与条件三等价

一般来说,统计方案数可以考虑按位从左到右dp,下标需要记录

另外,条件一的单调不降需要记录当前位的$a_i$ 的值

而条件四需要记录之前所有$a_i$ 的和

这样就至少需要记三维的状态了,但$n$ 有五千,显然不能直接dp

考虑通过差分去掉单调性

单调不降表示$a_{i+1}\ge a_i$ 也即 $a_{i+1}-a_i\ge 0$ ,于是令$d_i=a_i-a_{i-1}$ (先将$a_0$ 视为$0$ )

发现由于$a_1\ge 1$ ,$d_1$ 的限制有些另类,方便起见,还是将$a_0$ 视为 $1$ 吧

于是有$a_i=a_0+{\sum }_{j=1}^i{d}_j=1+{\sum }_{j=1}^i{d}_j$

将上式代入条件一,二,四,并化简得到新的三个条件
$$\forall i\in [1,n],d_i\ge 0$$

$$\sum _{i=1}^n{d}_i\le n-1$$

$$\begin{gathered} \sum _{j=1}^n{d}_j{c}_j\le 0 \\ {c}_j=\{\begin{array}{rcl}j-2& & 1\le j\le \lfloor \frac{n}{2}\rfloor +1\\ n-j+1& & \lfloor \frac{n}{2}\rfloor +1<j\le n\end{array} \end{gathered}$$

那么问题转化为统计合法的数列$d$ 的个数

发现条件二和条件三还是要记两个状态,不能直接做

观察$c_j$,发现先递增,后递减,并且除了$c_1=-1<0$,其他的$c_j$ 都非负,只有第一位最特殊

于是考虑提出 $d_1$

首先 $d_1\ge 0$ 是肯定的

将条件二中的 $d_1$ 提出得
$$d_1\le n-1-\sum _{i=2}^n{d}_i$$
将条件三中的 $d_1$ 提出得
$$d_1\ge \sum _{j=2}^n{d}_j{c}_j$$
发现这两个条件恰好就是 $d_1$ 的上下限,于是合起来就是
$$0\le \sum _{j=2}^n{d}_j{c}_j\le d_1\le n-1-\sum _{i=2}^n{d}_i$$
所以$d_1$ 可以取$[{\sum }_{j=2}^n{d}_j{c}_j,n-1-{\sum }_{j=2}^n{d}_j]$ 中的任意值

综上,当其他$d_i$ 都确定下来后,$d_1$ 有$max(n-{\sum }_{j=2}^n{d}_j(1+c_j),0)$ 种取值

然后数列就可以分成 $d_1$ 和 $d_2..d_n$ 两段来考虑了

考虑枚举后面那段的权值和 $i={\sum }_{j=2}^n{d}_j(1+c_j)$

易得,先考虑后面那段的合法方案 ,再考虑 $d_1$ 的取值就可以正确地算出总方案数

(对后面那段的限制只有 $d_i\ge 0$ ,其他所有限制都在 $d_1$ 的取值里考虑进去了)

令$f_i$ 为 $i={\sum }_{j=2}^n{d}_j(1+c_j)$ 的方案数

于是答案为${\sum }_{i=0}^n{f}_i\ast max(n-i,0)={\sum }_{i=0}^n{f}_i(n-i)$

问题转化为求$f_i$

令$dp[i][j]$ 表示后只考虑 $d_i..d_n$ 的情况下, $j={\sum }_{k=i}^n{d}_k(1+c_k)$ 的方案数

于是 $dp[2][1..n]$ 即为 $f[1]..f[n]$

大力转移即可,第一维可以滚动,也可以像完全背包那样优化掉

复杂度$O(n^2)$

代码

#include <bits/stdc++.h>
using namespace std;
int dp[5005],c[5005];
int n,MOD,ans;
int main()
{
    cin>>n>>MOD;
    for (int i=1;i<=n;++i) if (i<=n/2+1) c[i]=i-2; else c[i]=n-i+1;
    dp[0]=1;
    for (int i=n;i>1;--i)
        for (int j=c[i]+1;j<=n;++j)
            (dp[j]+=dp[j-c[i]-1])%=MOD;
    for (int i=0;i<=n;++i) ans=(1LL*dp[i]*(n-i)+ans)%MOD;
    cout<<ans<<endl;
    return 0;
}