题意:直线上有等间距的N个点,每个点有一个分数,落在一个点上能获得该点的分数。从第一个点跳到第N个点,每次可以向前跳1、2、3、4步,每种步距可以使用的次数有限,且保证用完所有的会跳到第N个点。求最大分数。N<=350,每种步距的使用次数<=40。
枚举步距的排列是阶乘级别的,但是,注意到只要每种步距已使用的次数确定,位置就确定了。因此,可以以每种步距已使用的次数为状态。
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAX_N = 350;
int N, M, a[MAX_N], t[4], f[41][41][41][41];
inline void upd(int& x, int v)
{
x = max(x, v);
}
int main()
{
scanf("%d %d", &N, &M);
for (int i = 0; i < N; ++i)
scanf("%d", &a[i]);
for (int i = 0; i < M; ++i) {
int b;
scanf("%d", &b);
++t[b-1];
}
f[0][0][0][0] = a[0];
for (int x = 1; x < N; ++x)
for (int i = 0; i <= t[1] && i*2 <= x; ++i)
for (int j = 0; j <= t[2] && i*2 + j*3 <= x; ++j)
for (int k = 0; k <= t[3] && i*2 + j*3 + k*4 <= x; ++k) {
int l = x - i*2 - j*3 - k*4;
if (l > t[0])
continue;
int& now = f[l][i][j][k];
if (l)
upd(now, f[l-1][i][j][k]);
if (i)
upd(now, f[l][i-1][j][k]);
if (j)
upd(now, f[l][i][j-1][k]);
if (k)
upd(now, f[l][i][j][k-1]);
now += a[x];
}
printf("%d\n", f[t[0]][t[1]][t[2]][t[3]]);
return 0;
}