SWJTU oj 2387 Magic Maze(DAG 求最長路)

這個題是給出一個有向無環圖。

那麼就考慮一個DP,dp[i]表示 i 結尾的路徑最大值。

因爲圖是有向無環圖,那麼至少存在一個起點和終點。

對於起點 u 來說:dp[u]=0;

那麼就倒着推,對於一個普通點 u ,可以又 u 的前驅點轉移而來,一步步推到起點。

那麼就記憶化搜索來寫,很簡單。

#include<stdio.h>
#include<algorithm>
#include<string>
#include<string.h>
#include<queue>
#include<vector>
#include<stack>
#include<math.h>
#include<map>
#include<set>
#include<iostream>
using namespace std;
#define LL long long
#define maxn 1005
struct node
{
    int pre;
    int len;
};
vector<node> v[maxn];
int n,m;
int dp[maxn];
void inti()
{
    memset(dp,0,sizeof(dp));
    for(int i=0;i<n;i++)
        v[i].clear();
}
int dfs(int u)
{
    if(dp[u]>0)
        return dp[u];
    int l=v[u].size();
    int ans=0;
    for(int i=0;i<l;i++)
    {
        node tmp=v[u][i];
        int pre=tmp.pre;
        int len=tmp.len;
        ans=max(ans,dfs(pre)+len);
    }
    dp[u]=ans;
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        inti();
        while(m--)
        {
            int u,vv,len;
            scanf("%d%d%d",&u,&vv,&len);
            node tmp;
            tmp.pre=u;tmp.len=len;
            v[vv].push_back(tmp);
        }
        for(int i=0;i<n;i++)
        {
            if(dp[i]>0)
                continue;
            dp[i]=dfs(i);
        }
        int ans=0;
        for(int i=0;i<n;i++)
            ans=max(ans,dp[i]);
        printf("%d\n",ans);
    }
    return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章