学不会的DP acwing 271. 杨老师的照相排列 [线性DP]

题目

有 NN 个学生合影，站成左端对齐的 kk 排，每排分别有 N1,N2,…,NkN1,N2,…,Nk 个人。 (N1≥N2≥…≥NkN1≥N2≥…≥Nk)

第 11 排站在最后边，第 kk 排站在最前边。

学生的身高互不相同，把他们从高到底依次标记为 1,2,…,N1,2,…,N。

在合影时要求每一排从左到右身高递减，每一列从后到前身高也递减。

问一共有多少种安排合影位置的方案？

下面的一排三角矩阵给出了当 N=6,k=3,N1=3,N2=2,N3=1N=6,k=3,N1=3,N2=2,N3=1 时的全部 1616 种合影方案。注意身高最高的是 11，最低的是 66。

123 123 124 124 125 125 126 126 134 134 135 135 136 136 145 146
45  46  35  36  34  36  34  35  25  26  24  26  24  25  26  25
6   5   6   5   6   4   5   4   6   5   6   4   5   4   3   3

输入格式

输入包含多组测试数据。

每组数据两行，第一行包含一个整数 kk 表示总排数。

第二行包含 kk 个整数，表示从后向前每排的具体人数。

当输入 k=0k=0 的数据时，表示输入终止，且该数据无需处理。

输出格式

每组测试数据输出一个答案，表示不同安排的数量。

每个答案占一行。

数据范围

1≤k≤51≤k≤5,学生总人数不超过 3030 人。

输入样例：

1
30
5
1 1 1 1 1
3
3 2 1
4
5 3 3 1
5
6 5 4 3 2
2
15 15
0

输出样例：

1
1
16
4158
141892608
9694845

代码

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
//#include 
//#include 
#define pb          push_back
#define ppb         pop_back
#define lbnd        lower_bound
#define ubnd        upper_bound
#define endl        '\n'
#define mll         map<ll,ll>
#define msl         map<string,ll>
#define mls         map<ll, string>
#define rep(i,a,b)  for(ll i=a;i<b;i++)
#define repr(i,a,b) for(ll i=b-1;i>=a;i--)
#define trav(a, x)  for(auto& a : x)
#define pll         pair<ll,ll>
#define vl          vector<ll>
#define vll         vector<pair<ll, ll>>
#define vs          vector<string>
#define all(a)      (a).begin(),(a).end()
#define F           first
#define S           second
#define sz(x)       (ll)x.size()
#define hell        1000000007
#define DEBUG       cerr<<"/n>>>I'm Here<<<<endl;
#define display(x)  trav(a,x) cout<<a<<" ";cout<<endl;
#define what_is(x)  cerr << #x << " is " << x << endl;
#define ini(a)      memset(a,0,sizeof(a))
#define ini2(a,b)   memset(a,b,sizeof(a))
//#define rep(i,a,b)  for(int i=a;i<=b;i++)
#define case        ll T;read(T);for(ll Q=1;Q<=T;Q++)
#define lowbit(x)   x&(-x)
#define pr          printf
#define sc          scanf
//#define _           0
#define ordered_set tree<ll, null_type,less<ll>, rb_tree_tag,tree_order_statistics_node_update>
#define FAST ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define DBG(x) \
    (void)(cout << "L" << __LINE__ \
    << ": " << #x << " = " << (x) << '\n')
#define TIE \
    cin.tie(0);cout.tie(0);\
    ios::sync_with_stdio(false);
//#define long long int

//using namespace __gnu_pbds;

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double PI    = acos(-1.0);
const double eps   = 1e-6;
const int    INF   = 0x3f3f3f3f;
const ll     LLINF = 0x3f3f3f3f3f3f3f3f;

template <typename T>
void read(T &x) {
    x = 0;
    int f = 1;
    char ch = getchar();
    while (!isdigit(ch)) {
        if (ch == '-') f = -1;
        ch = getchar();
    }
    while (isdigit(ch)) {
        x = x * 10 + (ch ^ 48);
        ch = getchar();
    }
    x *= f;
    return;
}

inline void write(ll x) {
    if(x<0) putchar('-'), x=-x;
    if(x>9) write(x/10);
    putchar(x%10+'0');
}

const ll     LN    = 5;
const int    maxn  = 205;
const int    N     = 31;

ll n;
ll f[N][N][N][N][N], arr[N];

void solve() {
	while (cin>>n && n!=0) {
		ini(f), ini(arr);
		for (ll i=1; i<=n; i++) cin>>arr[i];
		f[0][0][0][0][0] = 1;
		
		for (ll a = 0; a <= arr[1]; a ++ )
	            for (ll b = 0; b <= arr[2]; b ++ )
	                for (ll c = 0; c <= arr[3]; c ++ )
	                    for (ll d = 0; d <= arr[4]; d ++ )
	                        for (ll e = 0; e <= arr[5]; e ++ )
	                        {
	                            ll &x = f[a][b][c][d][e];
	                            if (a && a - 1 >= b) x += f[a - 1][b][c][d][e];
	                            if (b && b - 1 >= c) x += f[a][b - 1][c][d][e];
	                            if (c && c - 1 >= d) x += f[a][b][c - 1][d][e];
	                            if (d && d - 1 >= e) x += f[a][b][c][d - 1][e];
	                            if (e) x += f[a][b][c][d][e - 1];
	                        }
		cout<<f[arr[1]][arr[2]][arr[3]][arr[4]][arr[5]]<<endl;
	}
	
}

int main()
{
//	TIE;
//    #ifndef ONLINE_JUDGE
//    freopen ("input.txt","r",stdin);
//    #else
//    #endif
	solve();
//    case{solve();}
//    case{cout<<"Case "<
//	return ~~(0^_^0);
}