POJ2279 Mr. Young's Picture Permutations 线性DP

POJ2279 Mr. Young's Picture Permutations

给定一个三角形，将n个身高不同的人按照左低右高、前低后高的排列，且行列都满足递增条件，求组成这样的三角形的个数

dp[a][b][c][d][e] 表示第1排a人2排b人3排c人4排d人5排e人的方案数,每次加人的可行方案都是在原有队形轮廓上

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
int n,A[6];
vector<vector<vector<vector<vector<long long> > > > > dp;
long long dfs(int a,int b,int c,int d,int e)
{
    if(dp[a][b][c][d][e]>0) return dp[a][b][c][d][e];
    long long ans=0;
    if(a>0&&a>b) ans+=dfs(a-1,b,c,d,e);
    if(b>0&&b>c) ans+=dfs(a,b-1,c,d,e);
    if(c>0&&c>d) ans+=dfs(a,b,c-1,d,e);
    if(d>0&&d>e) ans+=dfs(a,b,c,d-1,e);
    if(e>0) ans+=dfs(a,b,c,d,e-1);
    return dp[a][b][c][d][e]=ans;
}
vector<long long>b;
vector<vector<long long> >c;
vector<vector<vector<long long> > >d;
vector<vector<vector<vector<long long> > > >e;
void getsize()
{
    for(int i1=0;i1<=A[1];++i1)
    {
        e.clear();
        for(int i2=0;i2<=A[2];++i2)
        {
            d.clear();
            for(int i3=0;i3<=A[3];++i3)
            {
                c.clear();
                for(int i4=0;i4<=A[4];++i4)
                {
                    b.clear();
                    for(int i5=0;i5<=A[5];++i5)
                    {
                        b.push_back(0ll);
                    }
                    c.push_back(b);
                }
                d.push_back(c);
            }
            e.push_back(d);
        }
        dp.push_back(e);
    }
    dp[0][0][0][0][0]=1ll;
}
int main()
{
    while(~scanf("%d",&n))
    {
        if(n==0) break;
        memset(A,0,sizeof(A));dp.clear();
        for(int i=1;i<=n;++i) scanf("%d",&A[i]);
        getsize();
        printf("%lld\n",dfs(A[1],A[2],A[3],A[4],A[5]));
    }
    return 0;
}