蓝桥杯 波动数列 DP背包

点击打开链接

x为首项p=a||p=-b
x+(x+p)+(x+2p)+...(x+(n-1)p)=s
nx=s-(p+2p+..(n-1)p)
a,b的个数和为 (1+(n-1))*(n-1)/2= n*(n-1)/2
此时枚举a,就能确定b得到
nx=s-ca+(n*(n-1))/2-c)*b
若t等于s-ca+(n*(n-1))/2-c)*b&&t%n==0  则首项确定 但是对应的c还有不同的组合形式


令c为背包体积,设d[i][j]:前i项体积为j的方法数 d[i][j]=d[i-1][j]+d[i-1][j-i] 第i项的体积(系数)为i  

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N=2e3+20;
const int mod=1e8+7;
ll dp[N*N];//
void DP(int n)
{
	dp[0]=1;
	for(int i=1;i<n;i++)//总共n-1项 
	{
		for(int j=i*(i+1)/2;j>=0;j--)//1+..i=(1+i)i/2 
		{
			if(j>=i)
			dp[j]=(dp[j-i]+dp[j])%mod; 
			else
			dp[j]=dp[j]%mod;
		}
	}
		
} 
int main()
{
	ll n,s,a,b;
	memset(dp,0,sizeof(dp));
	cin>>n>>s>>a>>b;  
	DP(n);
	ll ans=0;
	for(int i=0;i<=n*(n-1)/2;i++)
	{
		ll t=s-i*a+(n*(n-1)/2-i)*b;
		if(t%n==0)
		{
			ans=(ans+dp[i])%mod;	
		}	
	}	
	cout<<ans<<endl;
	return 0;
}