原题传送门
这题刚拿到的时候,反复揣%,但是并没有想出一个比较好的解法,果然是太弱了,于是一位大佬忍不住点开了题解,然后默默的给我指了指数据范围(K<=5)
然后这题就变成了一道记忆化搜索。。。
大力设状态,最多有5排,设dp [ i ] [ j ] [ k ] [ l ] [ p ]表示每一排人数分别为i,j,k,l,p时的种数
#include<bits/stdc++.h>
const int S=30;
int n,a[S];
unsigned f[S][S][S][S][S];
inline int dfs(int i,int j,int k,int l,int p)
{
if (i+j+k+l+p==0) return 1;
unsigned &tp=f[i][j][k][l][p];
if (tp) return tp;
if (i>j) tp+=dfs(i-1,j,k,l,p);
if (j>k) tp+=dfs(i,j-1,k,l,p);
if (k>l) tp+=dfs(i,j,k-1,l,p);
if (l>p) tp+=dfs(i,j,k,l-1,p);
if (p) tp+=dfs(i,j,k,l,p-1);
return tp;
}
int main()
{
scanf("%d",&n);
for (int i=0;i<n;i++)
scanf("%d",a+i);
printf("%u\n",dfs(a[0],a[1],a[2],a[3],a[4]));/* 传址输出 */
return 0;
}