洛谷P2132 小Z的队伍排列【记忆化搜索DP】【黄】

Date：2022.03.23
题目描述
小Z想给班里的同学拍一张合影，为此需要先让大家排好队伍。他希望大家站成 k 排，并规定了每排的人数，保证每一排的人数都不多于后面一排的人数。
这时小Z发现队伍看起来还是乱糟糟的，原因是大家的身高互不相同。于是，他希望排头对齐，每位同学都比自己正后方的同学以及排头方向的同学矮。
排完以后，善于思考的小Z还想知道一共有多少种排法。
例如，大家排成 3 排，且从后往前每排分别是 3、2、1 人，就有以下 16 种排法（每个数代表将所有同学身高从高到低排序后该同学的排名）：

可是班里一共有 n 个人，小Z算不出来了，希望你帮帮他。
输入格式
第一行包含一个整数 k。
第二行包含 k 个整数，表示从后往前每排的人数。
输出格式
一行，包含一个整数，表示队伍排列的方案数。
输入输出样例
输入 #1复制
3
3 2 1
输出 #1复制
16
输入 #2复制
5
1 1 1 1 1
输出 #2复制
1
说明/提示
【数据规模】
对于 30% 的数据，n ≤ 10。
对于另外 30% 的数据，只有最后一排超过 1 个人。
对于 100% 的数据，1 ≤ k ≤ 5，1 ≤ n ≤ 30，方案数小于 2^32。

思路：$f[i][j][k][p][q]:$第1~5行分别摆好前$i、j、k、p、q$个元素的方案数。
状态转移：因为前面的已固定，放数一定是在末尾（否则放进的元素会>这一行中前面的），因此可有分别放到五行的末尾这五种情况。
代码如下：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
LL f[31][31][31][31][31];
int n,m,a[7];
LL dfs(int i,int j,int k,int p,int q)
{
	if (i+j+k+p+q==0) return 1;
	LL &v=f[i][j][k][p][q];
	if (v) return v;
	if (i>j) v+=dfs(i-1,j,k,p,q);
	if (j>k) v+=dfs(i,j-1,k,p,q);
	if (k>p) v+=dfs(i,j,k-1,p,q);
	if (p>q) v+=dfs(i,j,k,p-1,q);
	if (q) v+=dfs(i,j,k,p,q-1);
	return v;
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>a[i];
    f[0][0][0][0][0]=1;
    printf("%lld",dfs(a[0],a[1],a[2],a[3],a[4]));
    return 0;
}