java蓝桥杯练习 波动数列

java蓝桥杯练习 波动数列

资源限制
时间限制：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。

//java code

import java.util.*;

public class Main {
    static int sum[];
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n=sc.nextInt();
        int s=sc.nextInt();
        int a=sc.nextInt();
        int b=sc.nextInt();
        int M=n*(n-1)/2;
        long result=0;
        sum=new int[M+1];
        dfs(n);
        long t=1l*a*M-s;
        if(t%n==0)
            result++;
        for (int i = 1; i <= M; i++) {
            t=t-a-b;
            if(t%n==0)
                result=(result+sum[i])%100000007;
        }
        System.out.println(result);
    }
    static void dfs(int i){
        sum[0]=1;
        for (int j = 1; j < i; j++) {
            for (int k = j*(j+1)/2; k >=j ; k--) {
                sum[k]=(sum[k]+sum[k-j])%100000007;
            }
        }
    }
}