P1541 乌龟棋【记忆化搜索】

题目背景

小明过生日的时候，爸爸送给他一副乌龟棋当作礼物。

题目描述

乌龟棋的棋盘是一行NN个格子，每个格子上一个分数（非负整数）。棋盘第1格是唯一的起点，第NN格是终点，游戏要求玩家控制一个乌龟棋子从起点出发走到终点。

乌龟棋中MM张爬行卡片，分成4种不同的类型（MM张卡片中不一定包含所有44种类型的卡片，见样例），每种类型的卡片上分别标有1,2,3,41,2,3,4四个数字之一，表示使用这种卡片后，乌龟棋子将向前爬行相应的格子数。游戏中，玩家每次需要从所有的爬行卡片中选择一张之前没有使用过的爬行卡片，控制乌龟棋子前进相应的格子数，每张卡片只能使用一次。

游戏中，乌龟棋子自动获得起点格子的分数，并且在后续的爬行中每到达一个格子，就得到该格子相应的分数。玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的分数总和。

很明显，用不同的爬行卡片使用顺序会使得最终游戏的得分不同，小明想要找到一种卡片使用顺序使得最终游戏得分最多。

现在，告诉你棋盘上每个格子的分数和所有的爬行卡片，你能告诉小明，他最多能得到多少分吗？

输入格式

每行中两个数之间用一个空格隔开。

第11行22个正整数N,MN,M，分别表示棋盘格子数和爬行卡片数。

第22行NN个非负整数，a_1,a_2,…,a_Na1​,a2​,…,aN​，其中a_iai​表示棋盘第ii个格子上的分数。

第33行MM个整数，b_1,b_2,…,b_Mb1​,b2​,…,bM​，表示M张爬行卡片上的数字。

输入数据保证到达终点时刚好用光MM张爬行卡片。

输出格式

11个整数，表示小明最多能得到的分数。

输入输出样例

输入 #1复制

9 5
6 10 14 2 8 8 18 5 17
1 3 1 2 1

输出 #1复制

73

思路

　　裸的记忆化搜索，开四维记录一下每张牌的用量然后dfs就可以了

CODE

#include <bits/stdc++.h>
#define dbg(x) cout << #x << "=" << x << endl
#define eps 1e-8
#define pi acos(-1.0)

using namespace std;
typedef long long LL;

const int inf = 0x3f3f3f3f;

template<class T>inline void read(T &res)
{
    char c;T flag=1;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}

namespace _buff {
    const size_t BUFF = 1 << 19;
    char ibuf[BUFF], *ib = ibuf, *ie = ibuf;
    char getc() {
        if (ib == ie) {
            ib = ibuf;
            ie = ibuf + fread(ibuf, 1, BUFF, stdin);
        }
        return ib == ie ? -1 : *ib++;
    }
}

int qread() {
    using namespace _buff;
    int ret = 0;
    bool pos = true;
    char c = getc();
    for (; (c < '0' || c > '9') && c != '-'; c = getc()) {
        assert(~c);
    }
    if (c == '-') {
        pos = false;
        c = getc();
    }
    for (; c >= '0' && c <= '9'; c = getc()) {
        ret = (ret << 3) + (ret << 1) + (c ^ 48);
    }
    return pos ? ret : -ret;
}

const int maxn = 1e5 + 7;

int n, m;

int a[maxn];
int b[maxn];

int f[50][50][50][50];

map<int, int> mp;

int dfs(int n1, int n2, int n3, int n4) {
    if(f[n1][n2][n3][n4]) {
        return f[n1][n2][n3][n4];
    }
    if(n1 == mp[1] && n2 == mp[2] && n3 == mp[3] && n4 == mp[4]) {
        return 0;
    }
    int pos = n1 + n2 * 2 + n3 * 3 + n4 * 4 + 1;
    int ans = 0;
    if(n1 < mp[1]) {
        ans = max(ans, dfs(n1 + 1, n2, n3, n4) + a[pos + 1]);
    }
    if(n2 < mp[2]) {
        ans = max(ans, dfs(n1, n2 + 1, n3, n4) + a[pos + 2]);
    }
    if(n3 < mp[3]) {
        ans = max(ans, dfs(n1, n2, n3 + 1, n4) + a[pos + 3]);
    }
    if(n4 < mp[4]) {
        ans = max(ans, dfs(n1, n2, n3, n4 + 1) + a[pos + 4]);
    }
    return f[n1][n2][n3][n4] = ans;
}

int main()
{
    read(n);
    read(m);
    for ( int i = 1; i <= n; ++i ) {
        read(a[i]);
    }
    for ( int i = 1; i <= m; ++i ) {
        read(b[i]);
        mp[b[i]]++;
    }
    int ans = dfs(0, 0, 0, 0);
    printf("%d
",ans + a[1]);
    return 0;
}