271. 杨老师的照相排列

最新推荐文章于 2024-10-02 08:00:00 发布

满地都是六便士他却抬头看向了月亮

最新推荐文章于 2024-10-02 08:00:00 发布

阅读量119

点赞数

分类专栏：动态规划笔记文章标签： c++ 算法线性dp

本文链接：https://blog.csdn.net/m0_46241477/article/details/124781068

版权

笔记同时被 2 个专栏收录

42 篇文章 1 订阅

订阅专栏

动态规划

21 篇文章 0 订阅

订阅专栏

给定学生总数N、排数k以及每排的人数，计算满足每排和每列身高递减的合影方案数。题目提供了一种线性DP的解法，通过遍历每个学生可能站立的位置，累加所有合法的排列方案。对于每组测试数据，输出不同的合影安排数量。代码示例中展示了如何实现这个算法，求解不同情况下的方案数。

摘要由CSDN通过智能技术生成

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

第 1 排站在最后边，第 k 排站在最前边。

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

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

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

下面的一排三角矩阵给出了当 N=6,k=3,N1=3,N2=2,N3=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

输入格式

输入包含多组测试数据。

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

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

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

输出格式

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

每个答案占一行。

数据范围

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

输入样例：

输出样例：

分析：这道题因为最多只有五排，我们就可以遍历每个人来了的时候可以站的位置，人来的顺序是按身高顺序，来之后我们判断它能去的排，所有方案相加即可，线性dp的思想。

代码如下：

#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 31;

int n;
LL f[N][N][N][N][N];

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

        for (int a = 0; a <= s[0]; a ++ )
            for (int b = 0; b <= min(a, s[1]); b ++ )
                for (int c = 0; c <= min(b, s[2]); c ++ )
                    for (int d = 0; d <= min(c, s[3]); d ++ )
                        for (int e = 0; e <= min(d, s[4]); 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[s[0]][s[1]][s[2]][s[3]][s[4]] << endl;
    }

    return 0;
}