POJ 1724 记忆话搜索dp

计算一个有向图从 1 到 N 在满足费用不超过K的前提下最短路是多少。。。

就是dp 状态是 dp[i][j] i 是现在是第 i 个点 j 是这个点费用为 j ，里面包含的是最短路程。

一个简单的剪枝是存下当前能到 n 点的最短路程，在搜索的过程中路程已经超过了这个路程就不用继续搜索了。

#include <stdio.h>
#include <iostream>
#include <queue>
#include <algorithm>
#include <map>
#include <vector>
#include <cmath>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <fstream>
#include <set>
#include <stack>
#include <list>
using namespace std;

#define READ freopen("acm.in","r",stdin)
#define WRITE freopen("acm.out","w",stdout)
#define ll long long
#define ull unsigned long long 
#define uint unsigned int
#define PII pair<int,int>
#define PDD pair<double,double>
#define fst first
#define sec second
#define MS(x,d) memset(x,d,sizeof(x))
#define INF 0x3f3f3f3f
#define ALL(x) x.begin(),x.end()
#define PB push_back
#define MOD 99991
#define MAX 111111

int dp[111][11111];// cost dis
int K,N,R;
struct node 
{
    int to,dis,cost;
    bool operator < (const node &o) const
    {
        return dis<o.dis;
    }
};
vector<node> G[111];
int ans=INF;
void dfs(int u,int cost,int dis)
{
    //cout<<u<<endl;
    if(dis>=ans)
        return ;
    if(u==N)
    {
        //cout<<u<<" "<<cost<<" "<<dis<<endl;
        if(cost<=K&&dp[N][cost]>dis)
            dp[N][cost]=dis,ans=min(ans,dis);
        //cout<<dp[N][cost]<<" "<<K<<endl;
        return ;
    }
    if(cost<=K&&dp[u][cost]>dis)
    {
        //cout<<u<<" "<<cost<<endl;
        dp[u][cost]=dis;
        for(int i=0;i<G[u].size();i++)
        {
            node v=G[u][i];
            dfs(v.to,cost+v.cost,dis+v.dis);
        }
    }
}

int main()
{
    READ;
    while(scanf("%d%d%d",&K,&N,&R)!=EOF)
    {
        for(int i=0;i<111;i++)
            G[i].clear();
        MS(dp,0x3f);
        for(int i=0;i<R;i++)
        {
            int f,t,dis,cost;
            scanf("%d%d%d%d",&f,&t,&dis,&cost);
            G[f].PB((node){t,dis,cost});
        }
        //for(int i=1;i<=N;i++)
        //    sort(ALL(G[i]));
        //dp[1][0]=0;
        dfs(1,0,0);
        int ans=INF;
        for(int i=0;i<=K;i++)
            ans=min(ans,dp[N][i]);
        if(ans==INF)
            cout<<-1<<endl;
        else
            cout<<ans<<endl;
    }
    return 0;
}