波动数列背包版.java

package A类有价值的回顾的;

import java.util.Arrays;
import java.util.Scanner;

/* 注意：有的文章自己都搞不清楚，代码倒是能够AC，趁早换
 * 设数列首项为x，F(i)={a,-b},第i项加上F(i)为第i+1项。所以第2项可表示为x+F(1)，第2项可表示为x+F(1)+F(2)，
 * 第3项可表示为x+F(1)+F(2)+F(3)...........把第一项到第n项累加起来可得表达式
 * x*n + (n-1)F(1) + (n-2)F(2) + ....... +F(n-1) =s
 * 倒过来
 * 也就是 x*n + F(n-1) + ....... + (n-2)F(2) + (n-1)F(1) = s
 */
public class 波动数列背包版 {
    private static long n,s,a,b;
    private static int cnt,e=1;
    private static int[][] F = new int[2][1100*1100];
    private static void DP(){
        F[e][0]=1;//也就是说对于第一个F(),一个a都没有的情况有1种，竟然可以满分过，虽然同样代码后来10%就超时了，至少证明可以全对
        F[e][1]=1;//也就是说对于第一个F(),有一个a的情况有1种
        for(int i=2;i<n;i++){ //有体积为1到n-1的n-1种物品       j代表a的总个数，i代表F()前的权值
            e = 1 - e;
            for(int j=0;j<=i*(i+1)/2;j++){//背包容量从0到 i*(i+1)/2，最多是把i以及i前的都是a

                if(i>j)//F[][j] 指的是a的个数为j时的方案数
                    F[e][j] = F[1-e][j];//权值都比和大了，还加上他干什么
                else//如果权值比和小，可以娶她，也可以不娶她
                    F[e][j] = (F[1-e][j] + F[1-e][j-i])%100000007;//这个不能省，不知道为啥
            }                      
        }

    }


    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        s = sc.nextInt();
        a = sc.nextInt();
        b = sc.nextInt();

        long x;
        DP();//先列出公式，a与b一共要有n*(n-1)/2项
        for(int j=0;j<=n*(n-1)/2;j++){

            x = s - j*a + (n*(n-1)/2-j)*b;
            if(x%n == 0)
                cnt = (cnt + F[e][j])%100000007;
        }
        System.out.println(cnt);
    }
}
/*
 * 问题描述
　　观察这个数列：
　　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。

 */