欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

洛谷 P1077 摆花

程序员文章站 2022-03-20 22:09:27
[TOC] 题目 "P1077 摆花" 思路 $\text{DP}$ $\text{f[i][j]}$表示前$i$种花摆了$j$盆的方案数。 设$x$($x$的范围为$[0,a_i]$)为第$i$种花摆了多少盆,$k$($k$的范围为$[0,m x]$)为前面$i 1$种花摆了多少盆。枚举$x$,$ ......

目录


题目

p1077 摆花

思路

$\text{dp}$
$\text{f[i][j]}$表示前$i$种花摆了$j$盆的方案数。
设$x$($x$的范围为$[0,a_i]$)为第$i$种花摆了多少盆,$k$($k$的范围为$[0,m-x]$)为前面$i-1$种花摆了多少盆。枚举$x$,$k$。
所以$f[i][x+k]=f[i][x+k]+f[i-1][k]$

代码

#include<bits/stdc++.h>
using namespace std;
int n,m;
int f[101][101];
inline int read(){
    int x=0;bool f=0;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-')f=!f;c=getchar();}
    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
    return f?-x:x;
}

int main(){
    n=read(),m=read();
    for(int i=0;i<=n;++i){
        f[i][0]=1;
    }
    for(int i=1,num;i<=n;++i){
        num=read();
        for(int j=0;j<=num;++j){
            for(int k=0;k<=m-j;++k){
                if(j==0&&k==0) continue;
                f[i][j+k]+=f[i-1][k];
                f[i][j+k]%=1000007;
            }
        }
    }
    printf("%d\n",f[n][m]%1000007);
    return 0;
}