1. 程式人生 > 實用技巧 >試題 歷屆試題 波動數列(dp)

試題 歷屆試題 波動數列(dp)

傳送門 問題描述   觀察這個數列:
  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。 思路: 由題意,第一項是任意整數,我們設為x。每一項比前一項增加a或減少b,我們可以用di表示第i項為前一項+a或-b。 則可以列出式子 x+x+d1+x+d1+d2+ ..... x+d1+....+dn-1=s;
有nx+(n-1)d1+....+dn-1=s;///這裡為方便使下標和常數相同 可以寫為nx+d1+....+(n-1)dn-1=s; 那麼x=[s-(d1+....+(n-1)dn-1)]/n; 因為數列上的都為整數,那麼一定有s%n和(d1+....+(n-1)dn-1)%n相同 則問題轉化為求序列d1+....+(n-1)dn-1的總和取模n餘數與s取模n相同的方案數量。 開一個數組dp[i][j]代表前i項數列得到總和取模n等於j的方案數量 第i項的方案等於前i-1項總和+a或-b後取模值相同兩個選擇的方案累加得到; 所以我們需要使得前i-1項的總和加上第i項取模得到的等於j 設前i-1項總和為c,
1.有c+ai≡j(mod n),那麼c≡j-ai(mod n); 2.有c-bi≡j(mod n),那麼c≡j+bi(mod n); 所以有dp[i][j]=dp[i-1][(j+ai)%n]+dp[i-1][(j-bi)%n]; ///注意取正餘數和初始化dp[0][0]=1(取0項餘數為0方案數為1)
#include<bits/stdc++.h>
using namespace std;
int dp[1005][1005];
const int mod = 100000007;
int main(){
    ///得到正餘數(a%b+b)%b;
    int n,s,a,b;
    cin>>n>>s>>a>>b;
    dp[0][0]=1;
    for(int i=1;i<=n;i++){
        for(int j=0;j<n;j++){
            dp[i][j]=dp[i-1][((j-a*i)%n+n)%n]+dp[i-1][(j+b*i)%n];
        }
    }///在總長度是n,當前數是x,組合n-1個數取模n結果和s取模n結果一樣的方案數
    cout<<dp[n-1][s%n]<<endl;
    return 0;
}