fzu2092收集水晶 记忆化搜索

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
#include<stack>
using namespace std;
int dp[11][11][11][11][200+5];
int v[11][11][200+5],n,m;
char g[15][15];
int dx[]={0,0,-1,1,0};
int dy[]={1,-1,0,0,0};

int ok(int a1,int b1,int a2,int b2)
{
    if(a1<0||a1>=n||b1<0||b1>=m) return 0;
    if(a2<0||a2>=n||b2<0||b2>=m) return 0;
    if(g[a1][b1]=='#'||g[a2][b2]=='#') return 0;
    return 1;
}

int dfs(int a1,int b1,int a2,int b2,int t)
{
    if(t>200) return 0;
    if(dp[a1][b1][a2][b2][t]!=-1) return dp[a1][b1][a2][b2][t];

    int i,j,ta,tb,sa,sb,tmp,mx;
    mx=0;
    for(i=0;i<5;i++)
    {
        for(j=0;j<5;j++)
        {
            ta=a1+dx[i],tb=b1+dy[i];
            sa=a2+dx[j],sb=b2+dy[j];
            if(ok(ta,tb,sa,sb)==0) continue;
            tmp=dfs(ta,tb,sa,sb,t+1);
            mx=max(mx,tmp);
        }
    }
    dp[a1][b1][a2][b2][t]=mx;
    if(a1==a2&&b1==b2) dp[a1][b1][a2][b2][t]+=v[a1][b1][t];
    else dp[a1][b1][a2][b2][t]+=v[a1][b1][t]+v[a2][b2][t];
    return dp[a1][b1][a2][b2][t];
}

int main()
{
    int _,i,j,t,x,y,w,k;
    scanf("%d",&_);
    while(_--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<n;i++)
            scanf("%s",g[i]);
        scanf("%d",&k);
        memset(v,0,sizeof(v));
        for(i=0;i<k;i++)
        {
            scanf("%d%d%d%d",&t,&x,&y,&w);
            x--;y--;
            v[x][y][t]+=w;
        }
        memset(dp,0xff,sizeof(dp));
        printf("%d\n",dfs(0,0,0,0,0));
    }
    return 0;
}

你可能感兴趣的:(fzu2092收集水晶 记忆化搜索)