codeforces 721C 记忆化搜索 / dp

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/qq_24477135/article/details/52959963

题目链接：http://codeforces.com/contest/721/problem/C

题目大意：给你n个点，m条有向边，每条边的距离，问从1到n最多可以经过多少点是的总距离不超过t，按路径顺序输出这些点

思路：dp[i][j]表示到达第i个点经过了j个点所用的最少距离。从1开始记忆华搜索，注意剪枝和路径的记录。

#include <iostream>
#include <cstdio>
#include <string>
#include <cstring>
#include <cmath>
#include <stack>
#include <queue>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <stdlib.h>
#include <iomanip>
#include <fstream>

using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define maxn 5005
#define MOD 1000000007
#define mem(a , b) memset(a , b , sizeof(a))
#define LL long long
#define ULL unsigned long long
#define FOR(i , n) for(int i = 1 ;  i<= n ; i ++)
typedef pair<int , int> pii;
vector<pii>v[maxn];
int dp[maxn][maxn];
int dp2[maxn][maxn];
int vis[maxn];
int path[maxn];
int n , m , t , ans;

bool DFS(int root , int num , int w)
{
    if(dp[root][num] <= w) return false;
    dp[root][num] = w;
    if(root == n)
    {
        if(w <= t && ans < num)
        {
            ans = num;
            path[num] = root;
            return true;
        }
        return false;
    }
    int flag = false;
    int up = v[root].size();
    for(int i = 0 ; i < up ; i ++)
    {
        if(DFS(v[root][i].first , num + 1 , w + v[root][i].second))
        {
            path[num] = root;
            flag = true;             //不能直接返回，因为还可能有更优的
        }
    }
    return flag;
}


int main()
{
    while(scanf("%d %d %d" , &n , &m , &t) != EOF)
    {
        for(int i = 0 ; i <= n ; i ++)
        {
            v[i].clear();
            for(int j = 0 ; j <= n ;  j ++)
            {
                dp[i][j] = MOD;
            }
        }
        for(int i = 0 ; i <= n ; i ++) dp[i][0] = 0;
        dp[1][1] = dp[1][0] = 0;
        mem(vis , 0);
        mem(path , -1);
        int u , to , w;
        for(int i = 0 ; i < m ; i ++)
        {
            scanf("%d %d %d" , &u , &to , &w);
            v[u].push_back(make_pair(to , w));
        }
       // vis[1] ++ ;
        ans = -1; path[1] = 1;
        int up = v[1].size();
        for(int i = 0 ; i < up ; i ++)
        {
           // path[2] = v[1][i].first;
            DFS(v[1][i].first , 2 , v[1][i].second);
        }

        printf("%d\n" , ans);
        for(int i = 1 ; i <= ans ; i ++)
        {
            printf("%d" , path[i]);
            if(i != ans) printf(" ");
        }
        cout << endl;
    }
    return 0;
}