G-Human Pyramid（动态规划）

G-Human Pyramid（动态规划）
题意：
给一个h和s，h代表金字塔的高度（由h*(h+1)/2个人组成金字塔），s代表强壮的人的人数。强壮的人必须由在地面或者由强壮的人支撑。求有多少种安排？

思路：
如果一个位置是强壮的人，则其下方全是强壮的人。并且其右下方也为强壮的人。

dp[i][j][k]表示从右往左数第i列，总共使用了j个强壮的人，当前列使用k个强壮的人的方案数。

对于第p列使用j个强壮的人，则前一列至少使用j-1个人。

状态转移方程为：
dp[i][j][k]=dp[i][j][k+1]+dp[i-1][j-k][k-1];

代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int dp[105][5055][105];//dp[i][j][k]，i：第几列，j：使用多少个S，k当前列使用多少个S
const int mod=1e9+7;
int main(){
    int h,s;
    cin>>h>>s;
    memset(dp,0,sizeof(dp));
    dp[0][0][0]=1;
    for(int i=1;i<=h;i++){
        for(int j=0;j<=s;j++){
            for(int k=i;k>=0;k--){
                if(k==0)
                    dp[i][j][k]=(dp[i][j][k+1]+dp[i-1][j][k])%mod;
                else
                    dp[i][j][k]=(dp[i][j][k+1]+dp[i-1][j-k][k-1])%mod;
            }
        }
    }
    cout<<dp[h][s][0];
}