uva 11125 - Arrange Some Marbles(记忆化搜索）

题目链接：uva 11125 - Arrange Some Marbles

题目大意：给出n表示有n种颜色的大理石，然后a1~an，表示各种颜色的大理石有多少个，然后现在要将大理石排成一排，颜色相同的的为一组，相邻的两组颜色和大小不能相同，而且头尾两组的大小和颜色也不能相同，问说有多少排列的方式。

解题思路：一开始被题意搞蒙了，其实记忆化搜索就可以了，开一个5维数组，rec[bc][bn][x][y][t]，bc，bn分别为第一组的颜色和大小，x，y分别为前一组的颜色和大小，t为压缩的各种颜色大理石的余量。

#include <stdio.h>
#include <string.h>

const int N = 8;
const int MAXN = 4500;

int n, a
, bc, bn;
int rec

[MAXN];

int dp(int x, int y, int s) {

int t = 0;
for (int i = 0; i < 4; i++)
t = t * 8 + a[i];

if (!s) {
if (x != bc && y != bn) return 1;
else return 0;
}

int& ans = rec[bc][bn][x][y][t];
if (ans != -1) return ans;
ans = 0;

for (int i = 0; i < n; i++) {
if (i == x) continue;
for (int j = 1; j <= 3 && j <= a[i]; j++) {
if (j == y) continue;
a[i] -= j;
ans += dp(i, j, s - j);
a[i] += j;
}
}
return ans;
}

int solve() {
int s = 0, ans = 0;
scanf("%d", &n);
memset(a, 0, sizeof(a));
for (int i = 0; i < n; i++) {
scanf("%d", &a[i]);
s += a[i];
}

if (!s) return 1;

for (int i = 0; i < n; i++) {
for (int j = 1; j <= 3 && j <= a[i]; j++) {
bc = i, bn = j;
a[i] -= j;
ans += dp(i, j, s - j);
a[i] += j;
}
}
return ans;
}

int main () {
int cas;
scanf("%d", &cas);
memset(rec, -1, sizeof(rec));
while (cas--) {
printf("%d\n", solve());
}
return 0;
}