蓝桥杯波动数列JAVA代码

转载：吴泽雄的博客

问题描述
　　观察这个数列：
　　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。
　思路：用for循环，对首数-9到9进行遍历， 再利用递归，在后面的数进行加a，减b处理。
　代码：

import java.util.Scanner;

public class Main{
        static int n;
        static int s;
        static int a;
        static int b;
        static int count;

        public static void main(String[] args){
            // 输入数据
            Scanner s1 = new Scanner(System.in);
            Scanner s2 = new Scanner(s1.nextLine());
            int arr[] = new int[4];
            int len = 0;
            while(s2.hasNext())
            {
                arr[len] = s2.nextInt();
                len++;
            }

            n = arr[0];
            s = arr[1];
            a = arr[2];
            b = arr[3];

            //首位从-9开始判断
            for(int i=-9; i<=9; i++)
            {
                fun(i,0,i);
            }

            System.out.println(count%100000007);
        }

        static int sum = 0;
        // i表示所处理的第一个元素标记，j表示数列所进行长度，
        public static void fun(int i,int j, int k)
        {
            if(i>9)
            {
                return;
            }
            //若到达末尾，判断和是否为s，方案加1，sum清0
            if(j==n)
            {
                if(sum == s)
                {
                    count++;
                }
                sum = 0;
                return ;
            }
            //k为每位的值
            sum = sum + k;//计算数组的和
            fun(i, j+1, k+a);
            fun(i, j+1, k-b);

        }
    }