POJ - 2279 Mr. Young‘s Picture Permutations（线性DP）

POJ - 2279 Mr. Young’s Picture Permutations

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

typedef long long LL;
int s[6];
int n;

int main()
{
	while(cin>>n&&n)
	{
		memset(s,0,sizeof s);
		for(int i=0;i<n;i++) cin>>s[i];
		
		LL f[s[0]+1][s[1]+1][s[2]+1][s[3]+1][s[4]+1];
		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<=s[1];b++)
				for(int c=0;c<=s[2];c++)
					for(int d=0;d<=s[3];d++)
						for(int e=0;e<=s[4];e++)
						{ 
							if(a&&a-1>=b) f[a][b][c][d][e]+=f[a-1][b][c][d][e];
							if(b&&b-1>=c) f[a][b][c][d][e]+=f[a][b-1][c][d][e];
							if(c&&c-1>=d) f[a][b][c][d][e]+=f[a][b][c-1][d][e];
							if(d&&d-1>=e) f[a][b][c][d][e]+=f[a][b][c][d-1][e];
							if(e) f[a][b][c][d][e]+=f[a][b][c][d][e-1];
						}
		cout<<f[s[0]][s[1]][s[2]][s[3]][s[4]]<<endl;
	}
	return 0;
}