题意
给出n行,每行有 N[i] N [ i ] 个人,我们要给他们安排位置,满足每一排从左到右身高递减,每一列从后到前身高也递减,求出所有方案。
思路
杨氏矩阵做法:
这样的要求满足杨氏矩阵的条件,所以我们就可以利用钩子公式求出每种固定形状的杨氏矩阵的排列数量:
ans=n!/(每个格子的下面和右边的格子数量+1)
a
n
s
=
n
!
/
(
每
个
格
子
的
下
面
和
右
边
的
格
子
数
量
+
1
)
。
动态规划做法:
n最多是5,我们就可以用五维的动态规划。
设f[a][b][c][d][e]代表第一行有a个,第二行有b个,第三行有c个……时的方案数,因为我们之前处理的答案都是满足条件的,所以我们可以从当前的答案转移到后面的答案:
如果a<n[1],则f[a+1][b][c][d][e]+=f[a][b][c][d][e]
如
果
a
<
n
[
1
]
,
则
f
[
a
+
1
]
[
b
]
[
c
]
[
d
]
[
e
]
+
=
f
[
a
]
[
b
]
[
c
]
[
d
]
[
e
]
如果b<n[2]且a>b,则f[a][b+1][c][d][e]+=f[a][b][c][d][e]
如
果
b
<
n
[
2
]
且
a
>
b
,
则
f
[
a
]
[
b
+
1
]
[
c
]
[
d
]
[
e
]
+
=
f
[
a
]
[
b
]
[
c
]
[
d
]
[
e
]
比较两行的长度以及当前这行有没有放完进行转移。
代码
杨氏矩阵:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,hang[31],hook[31],cnt;
long long dive,ans,gys;
int main() {
while (scanf("%d",&n)&&n) {
cnt=0;
for (int i=1;i<=n;i++)
scanf("%d",&hang[i]);
for (int i=1;i<=n;i++)
for (int j=1;j<=hang[i];j++) {
hook[++cnt]=hang[i]-j+1;//加上右边的格子数再+1
for (int k=i+1;k<=n;k++)
if (hang[k]>=j) hook[cnt]++;//加上下面的格子数
else break;
}
ans=dive=1;
for (int i=1;i<=cnt;i++) {
ans*=i;//n!
dive*=hook[i];//公式中的除数
gys=__gcd(ans,dive);
if (gys!=1) {//30!会超出longlong,我们要在乘的时候就判断约分
ans/=gys;
dive/=gys;
}
}
printf("%lld\n",ans/dive);
}
}
动态规划:
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,a[6],N[6];
int main() {
while (scanf("%d",&n)&&n) {
for (int i=1;i<=n;i++)
scanf("%d",&N[i]);
for (int i=n+1;i<=5;i++)
N[i]=0;
long long f[N[1]+1][N[2]+1][N[3]+1][N[4]+1][N[5]+1];//动态开数组,不然会MLE,不知为什么
memset(f,0,sizeof(f));
f[0][0][0][0][0]=1;
for (a[1]=0;a[1]<=N[1];a[1]++)
for (a[2]=0;a[2]<=N[2];a[2]++)
for (a[3]=0;a[3]<=N[3];a[3]++)
for (a[4]=0;a[4]<=N[4];a[4]++)
for (a[5]=0;a[5]<=N[5];a[5]++) {
if (a[1]<N[1]) f[a[1]+1][a[2]][a[3]][a[4]][a[5]]+=f[a[1]][a[2]][a[3]][a[4]][a[5]];
if (a[2]<N[2]&&a[1]>a[2]) f[a[1]][a[2]+1][a[3]][a[4]][a[5]]+=f[a[1]][a[2]][a[3]][a[4]][a[5]];
if (a[3]<N[3]&&a[2]>a[3]) f[a[1]][a[2]][a[3]+1][a[4]][a[5]]+=f[a[1]][a[2]][a[3]][a[4]][a[5]];
if (a[4]<N[4]&&a[3]>a[4]) f[a[1]][a[2]][a[3]][a[4]+1][a[5]]+=f[a[1]][a[2]][a[3]][a[4]][a[5]];
if (a[5]<N[5]&&a[4]>a[5]) f[a[1]][a[2]][a[3]][a[4]][a[5]+1]+=f[a[1]][a[2]][a[3]][a[4]][a[5]];
}//一堆判断转移
printf("%lld\n",f[N[1]][N[2]][N[3]][N[4]][N[5]]);
}
}