线性DP 杨老师的拍照排列

分析: 利用数组a记录所有已经安排好的学生,a数组的和代表了一个状态设为k,k -> k + 1符合线性dp的特点

思路: 开五维数组f[31][31][31][31][31][31], 边界f[0][0][0][0][0] = 1, 其余设为0, 目标f[N1][N2][N3][N4][N5], 若a1 < N1, 转移为f[a1 + 1][a2][a3][a4][a5] += f[a1][a2][a3][a4][a5], 其余四行一样例如第二行若a2 < N2 && a1 > a2 转移为f[a1][a2 + 1][a3][a4][a5] += f[a1][a2][a3][a4][a5].

Code：

#include <iostream>
#include <cstring>

using namespace std;

typedef long long ll;
const int N = 6, M = 31;
int a[N];

ll f[M][M][M][M][M];

int main()
{
    int n;
    while(cin >> n && n)
    {
        memset(a, 0, sizeof a);
        memset(f, 0, sizeof f);
        f[0][0][0][0][0] = 1;
        for(int i = 1; i <= n; i ++) cin >> a[i];
        
        for(int a1 = 0; a1 <= a[1]; a1 ++)
            for(int a2 = 0; a2 <= min(a1, a[2]); a2 ++)
                for(int a3 = 0; a3 <= min(a2, a[3]); a3 ++)
                    for(int a4 = 0; a4 <= min(a3, a[4]); a4 ++)
                        for(int a5 = 0; a5 <= min(a4, a[5]); a5 ++)
                        {
                            ll &x = f[a1][a2][a3][a4][a5];
                            if(a1 + 1 <= a[1]) f[a1 + 1][a2][a3][a4][a5] += x;
                            if(a1 > a2 && a2 + 1 <= a[2]) f[a1][a2 + 1][a3][a4][a5] += x;
                            if(a2 > a3 && a3 + 1 <= a[3]) f[a1][a2][a3 + 1][a4][a5] += x;
                            if(a3 > a4 && a4 + 1 <= a[4]) f[a1][a2][a3][a4 + 1][a5] += x;
                            if(a4 > a5 && a5 + 1 <= a[5]) f[a1][a2][a3][a4][a5 + 1] += x;
                        }
        cout << f[a[1]][a[2]][a[3]][a[4]][a[5]] << endl;
    }
    return 0;
}