首页 > 代码库 > poj 1695 Magazine Delivery 记忆化搜索

poj 1695 Magazine Delivery 记忆化搜索

dp[a][b][c],表示三个人从小到大依次在a,b,c位置时,距离结束最少的时间。

每次选一个人走到c+1位置搜索就好了。

坑点在于不能floyd,估计题目没说清楚,意思就是如果没送Li,那么Li~n的点连去都不能去。

#include<cstdio>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
#define INF 0x3f3f3f3f
int dp[31][31][31];
int mp[31][31];
int n;
int dfs(int a[])
{
    if(~dp[a[0]][a[1]][a[2]]) return dp[a[0]][a[1]][a[2]];
    if(a[2]==n) return 0;
    int tmp[3];
    int mins=INF;
    for(int i=0;i<3;i++)
    {
        tmp[0]=a[0];tmp[1]=a[1];tmp[2]=a[2];
        tmp[i]=a[2]+1;
        sort(tmp,tmp+3);
        mins=min(mins,dfs(tmp)+mp[a[i]][tmp[2]]);
    }
    return dp[a[0]][a[1]][a[2]]=mins;
}
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--)
    {
        scanf("%d",&n);
        memset(dp,-1,sizeof(dp));
        for(int i=1;i<=n;i++) mp[i][i]=1;
        for(int i=1;i<=n-1;i++)
        {
            for(int j=i+1;j<=n;j++)
            {
                scanf("%d",&mp[i][j]);
                mp[j][i]=mp[i][j];
            }
        }
//        for(int k=1;k<=n;k++)
//        {
//            for(int i=1;i<=n;i++)
//            {
//                for(int j=1;j<=n;j++)
//                {
//                    mp[i][j]=min(mp[i][j],mp[i][k]+mp[k][j]);
//                }
//            }
//        }
        int a[3]={1,1,1};
        printf("%d\n",dfs(a));
    }
    return 0;
}