波动数列（DP）

历届试题 波动数列  

时间限制：1.0s   内存限制：256.0MB

   

问题描述

　　观察这个数列：
　　1 3 0 2 -1 1 -2 ...

　　这个数列中后一项总是比前一项增加2或者减少3。

　　栋栋对这种数列很好奇，他想知道长度为 n 和为 s 而且后一项总是比前一项增加a或者减少b的整数数列可能有多少种呢？

输入格式

　　输入的第一行包含四个整数 n s a b，含义如前面说述。

输出格式

　　输出一行，包含一个整数，表示满足条件的方案数。由于这个数很大，请输出方案数除以100000007的余数。

样例输入

4 10 2 3

样例输出

2

样例说明

　　这两个数列分别是2 4 1 3和7 4 1 -2。

数据规模和约定

　　对于10%的数据，1<=n<=5，0<=s<=5，1<=a,b<=5；
　　对于30%的数据，1<=n<=30，0<=s<=30，1<=a,b<=30；
　　对于50%的数据，1<=n<=50，0<=s<=50，1<=a,b<=50；
　　对于70%的数据，1<=n<=100，0<=s<=500，1<=a, b<=50；
　　对于100%的数据，1<=n<=1000，-1,000,000,000<=s<=1,000,000,000，1<=a,b<=1,000,000。

本来想找出一些数学规律的，发现了n * m + a * x - b * y = s，x + y = （n-1）*n/2.

暴力循环写了一次，超时间了。

百度下来原来是DP，也是用的上面那个式子，但是预处理把x取各种值有多少种可能预处理出来了。

状态转移方程：

if i > j

    d(i,j) = d(i - 1, j);

else

    d(i, j) = d(i - 1, j) + d(i - 1, j - i);

d(i, j)表示前i个元素使用j个a的序列的总数量。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define N 1101
#define mod 100000007

int dp[2][N * N];
int it;
ll n, s, a, b;
int ans = 0;

int main()
{
    //freopen("input2.txt", "r", stdin);
    scanf("%lld%lld%lld%lld", &n, &s, &a, &b);

    ll all = (n - 1) * n / 2;

    dp[it][0] = 1;
    for(ll i = 1; i < n; i++){
        it ^= 1;
        for(ll j = 0; j <= i * (i + 1) / 2; j++){
            if(i > j){
                dp[it][j] = dp[it ^ 1][j];
            }else{
                dp[it][j] = (dp[it ^ 1][j] + dp[it ^ 1][j - i]) % mod;
            }
        }
    }

    for(ll i = 0; i <= all; i++){
        if((s - i * a + (all - i) * b) % n == 0){
            ans = (ans + dp[it][i]) % mod;
        }
    }
//cout << dp[it][all] << endl;
    printf("%d", ans);

    return 0;
}