poj-1724-ROADS(dfs)

题目地址

http://poj.org/problem?id=1724

题目大意

N个城市，编号1到N。城市间有R条单向道路。
每条道路连接两个城市，有长度和过路费两个属性。
Bob只有K块钱，他想从城市1走到城市N。问最短共需要走多长的路。如果到不了N，输
出-1

解题思路

• 深度优先搜索
• 剪枝
  
  • 1) 如果当前已经找到的最优路径长度为L ,那么在继续搜索的过程中，总长度已经大
    于L的走法，就可以直接放弃，不用走到底了
  • 2) 用midL[k][m] 表示：走到城市k时总过路费为m的条件下，最优路径的长度。若在
    后续的搜索中，再次走到k时，如果总路费恰好为m，且此时的路径长度已经超过
    midL[k][m],则不必再走下去了。

Code

#include <stdio.h>
#include <iostream>
#include <stdlib.h>
#include <string.h>
#include <queue>
#include <map>
#include <vector>
#include <math.h>
#include <algorithm>
#define INF 0x3fffffff
#define N 110

using namespace std;
typedef long long LL;

int k, n, r;

struct Node {
    int to;
    int len;
    int toll;
};

vector<Node> G[N];

int mark[N][11000];
bool vis[N];
int g_min = INF;
int cur_cost = 0;
int cur_toll = 0;

void dfs(int x) {
    if (x == n) {
        if (cur_toll <= k) {
            g_min = min(g_min, cur_cost);
            return ;
        } else {
            return ;
        }
    }

    for (int i = 0; i < G[x].size(); i++) {
        int d = G[x][i].to;
        if (!vis[d]) {
            // cut
            if (cur_toll + G[x][i].toll > k) continue;
            if (cur_cost + G[x][i].len > g_min) continue;
            if (cur_cost + G[x][i].len > mark[d][cur_toll+G[x][i].toll]) continue;
            mark[d][cur_toll+G[x][i].toll] = cur_cost + G[x][i].len;

            cur_cost += G[x][i].len;
            cur_toll += G[x][i].toll;
            vis[d] = 1;
            dfs(d);

            cur_cost -= G[x][i].len;
            cur_toll -= G[x][i].toll;
            vis[d] = 0;
        }
    }
}

int main() {

#ifndef ONLINE_JUDGE
    freopen("in", "r", stdin);
#else
    //
#endif

    while (cin >> k >> n >> r) {
        int s;
        Node node;
        for (int i = 0; i < r; i++) {
            cin >> s >> node.to >> node.len >> node.toll;
            if (s != node.to) {
                G[s].push_back(node);
            }
        }

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < 10000; j++) {
                mark[i][j] = INF;
            }
        }

        memset(vis, 0, sizeof(vis));
        g_min = INF;
        cur_cost = 0;
        cur_toll = 0;
        vis[1] = 1;
        dfs(1);
        if (g_min == INF) {
            cout << "-1" << endl;
        } else {
            cout << g_min << endl;
        }
    }
    return 0;
}

参考

https://d396qusza40orc.cloudfront.net/pkupop/lectures/Week15/W15-02_%E6%B7%B1%E6%90%9C%E4%B9%8B%E5%AF%BB%E8%B7%AF%E9%97%AE%E9%A2%98.pdf