藍橋杯波動數列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);

        }
    }