【ACWing】271. 杨老师的照相排列

题目地址：

https://www.acwing.com/problem/content/description/273/

有$N$个学生合影，站成左端对齐的$k$排，每排分别有$N_1,N_2,\dots ,N_k$个人。$N_1\ge N_2\ge \dots \ge N_k$。第$1$排站在最后边，第$k$排站在最前边。学生的身高互不相同，把他们从高到底依次标记为$1,2,\dots ,N$。在合影时要求每一排从左到右身高递减，每一列从后到前身高也递减。问一共有多少种安排合影位置的方案？下面的一排三角矩阵给出了当$N=6,k=3,N_1=3,N_2=2,N_3=1$时的全部$16$种合影方案。注意身高最高的是$1$，最低的是$6$。

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

输入格式：
输入包含多组测试数据。每组数据两行，第一行包含一个整数 $k$表示总排数。第二行包含$k$个整数，表示从后向前每排的具体人数。当输入$k=0$的数据时，表示输入终止，且该数据无需处理。

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

数据范围：
$1\le k\le 5$，学生总人数不超过$30$人。

思路是动态规划。设$f[x_0][x_1][x_2][x_3][x_4]$表示第$i$行摆放了$x_i$个人，并且摆放的这些人的身高是前$s=x_0+x_1+x_2+x_3+x_4$名的情况下的方案数，则初始条件$f[0][0][0][0][0]=0$。我们先证明这前$s$名的同学的摆放方案都是每行连续并且靠左的。可以用数学归纳法，首先考虑第$1$高的同学放在哪儿，显然只能放在第$0$行第$0$列，成立；假设对于第$k$高的同学也成立，考虑第$k+1$高的同学放在哪儿，他是不能站在左边有空位的位置的，否则那个空位没人可以满足。接下来考虑递推关系。只需要考虑第$s$名同学的摆放位置即可，他可以位于第$0,1,...,4$行，如果他位于第$0$行，说明$x_0\ge x_1+1$，并且计数加上$f[x_0-1][x_1][x_2][x_3][x_4]$；如果他位于第$1$行，说明$x_1\ge x_2+1$，并且计数加上$f[x_0][x_1-1][x_2][x_3][x_4]$，以此类推。能放在最后一行当且仅当$x_4>0$，计数加上$f[x_0][x_1][x_2][x_3][x_4-1]$。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

const int N = 31;
int n;
long f[N][N][N][N][N];

int main() {
  while (cin >> n, n) {
    int a[5] = {0};
    for (int i = 0; i < n; i++) cin >> a[i];
    memset(f, 0, sizeof f);
    f[0][0][0][0][0] = 1;

    for (int x0 = 0; x0 <= a[0]; x0++)
      for (int x1 = 0; x1 <= min(x0, a[1]); x1++)
        for (int x2 = 0; x2 <= min(x1, a[2]); x2++)
          for (int x3 = 0; x3 <= min(x2, a[3]); x3++)
            for (int x4 = 0; x4 <= min(x3, a[4]); x4++) {
              long &v = f[x0][x1][x2][x3][x4];
              if (x0 && x0 >= x1 + 1) v += f[x0 - 1][x1][x2][x3][x4];
              if (x1 && x1 >= x2 + 1) v += f[x0][x1 - 1][x2][x3][x4];
              if (x2 && x2 >= x3 + 1) v += f[x0][x1][x2 - 1][x3][x4];
              if (x3 && x3 >= x4 + 1) v += f[x0][x1][x2][x3 - 1][x4];
              if (x4) v += f[x0][x1][x2][x3][x4 - 1];
            }

    cout << f[a[0]][a[1]][a[2]][a[3]][a[4]] << endl;
  }

  return 0;
}

每组数据时空复杂度$O(\prod N_i)$。