AcWing 1214.波动数列

题目描述

原题链接：1214.波动数列

解题思路

1.分析时间复杂度

在题目中：1$\le n\le 1000$，$n^2=10^6<10^8$，因此本题的时间复杂度差不多$o(n^2)$。

2.数学推导

假设数列的第一个数为$x$，第$i$次操作的数为$d_i$，$d_i=+a$或$-b$。
那么一个长度为$n$的数列就是$x,x+d_1,x+d_1+d_2,...,x+d_1+d_2+...+d_{n-1}$。
这一串数列的和就是$nx+(n-1)d_1+(n-2)d_2+...+d_{n-1}=s$。
进行变换后可得，$x=\frac{s-(n-1)d_1-(n-2)d_2-...-d_{n-1}}{n}$
因为$x$是整数，所以$s-(n-1)d_1-(n-2)d_2-...-d_{n-1}$一定是$n$的倍数，即$s$与$(n-1)d_1+(n-2)d_2+...+d_{n-1}$模$n$同余。这其实是一个组合问题，在$n-1$个操作下，如何组合使得数列之和与$s$同余。

3.闫氏dp分析

假设f[i][j]表示在考虑第$i$次操作下，即第$i$次操作$+a$或$-b$，前$i+1$个数模$n$结果为$j$的方案数。
假如第$i$次操作是$+a$，那么前$i+1$个数字的和是$x+(n-1)d_1+(n-2)d_2+...+(n-(i-1))d_{i-1}+(n-i)a\equiv j(modn)$。
那么，$x+(n-1)d_1+(n-2)d_2+..+(n-(i-1)d_{i-1}\equiv j-(n-i)a(modn)$。上述左端的式子刚好为考虑$i-1$次操作之后数列的和。
通过以上的分析我们可以知道，当第$i$次操作为$+a$时，方案数为f[i-1][j-(n-1)*a]。同理，当第$i$次操作为$-b$时，方案数为f[i-1][j+(n-1)*b]。
即f[i][j] = f[i-1][j-(n-1)*a] + f[i-1][j+(n-1)*b]。

4.一些注意点

（1）由于$j-(n-1)\ast a$和$s$都有可能是负数，而数组下标只能大于等于0，因此在取模的时候要注意是结果必须大于等于0。
（2）关于初始化的问题，当$i=0$时，满足题意的方案只有一种，就是$x=s$，因此f[0][0] = 1。

c++代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1010;
const int P = 1e8+7;
int n,s,a,b,f[N][N];
int get_mod(int a, int b)
{
    return (a % b + b) % b; 
}
int main()
{
    cin >> n >> s >> a >> b;
    f[0][0] = 1;
    for(int i = 1; i < n; i++)
    {
        for(int j = 0; j < n; j++)
        {
            f[i][j] = (f[i-1][get_mod(j-(n-i)*a, n)] + f[i-1][get_mod(j+(n-i)*b, n)]) % P;
        }
    }
    cout << f[n-1][get_mod(s,n)];
}

参考文献

https://www.acwing.com/solution/content/9223/