摆花

题目链接:https://www.luogu.com.cn/problem/P1077

题意:

有n种花,每种花最多摆ai个,总共需要摆m个,求摆花的方案数

思路:

记dp[i,j] 为 以i 作结尾 摆了 j 个花的方案数

dp[i,j] = dp[i,j] + dp[i-1,z]

其中z指的是前面摆的花的总数

取模时,不能dp[i,j]+=dp[i-1,z]%mod

这会导致溢出,因为dp[i,j]没有取模

因此dp[i,j] = (dp[i,j]%mod + dp[i-1,z] %mod)%mod

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define endl "\n"
#define fi first
#define se second
#define int long long
//#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef __int128 lll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=1e5+5;
const int mod=1e6+7;

void solve(){
	int n,m;
	cin>>n>>m;
	vector<int>a(n+1,0);
	rep(i,1,n) cin>>a[i];
	vector<vector<int>>dp(n+1,vector<int>(m+1,0));
	for(int i=0;i<=a[1]&&i<=m;i++){
		dp[1][i]=1;
	}
	for(int i=2;i<=n;i++){
		for(int j=0;j<=a[i];j++){
			for(int z=0;j+z<=m;z++){
				dp[i][j+z]=(dp[i][j+z]%mod+dp[i-1][z]%mod)%mod;
			}
		}
	}
	cout<<dp[n][m]<<endl;
}

signed main()
{
	ios::sync_with_stdio(false),cin.tie(0);
	int T=1;
	
	while(T--){
	solve();
	}
	
	return 0;
}


posted @ 2025-03-04 19:29  Marinaco  阅读(18)  评论(0)    收藏  举报
//雪花飘落效果