A Walk Through the Forest HDU - 1142——(最短路+DAG+拓撲+DFS記憶化)

總結

跑短路,然後在DAG上進行DP,可以拓撲DP,但是要構造DAG圖,這題可以DFS記憶化,實際上,還是DP,代碼量確少了不是一點點。

題意:

一條路徑從1到2,每條邊權滿足dis[a]>dis[b],dis表示到2的最短距離,計算所有可能的路徑數

題目鏈接

//#pragma GCC optimize(2)
#include<bits/stdc++.h>
//typedef long long ll;
#define ull       unsigned long long
#define int       long long
#define F           first
#define S           second
#define endl        "\n"//<<flush
#define eps         1e-6
#define base        131
#define lowbit(x)   (x&(-x))
#define PI          acos(-1.0)
#define inf         0x3f3f3f3f
#define MAXN        0x7fffffff
#define INF         0x3f3f3f3f3f3f3f3f
#define ferma(a,b)  pow(a,b-2)
#define pb          push_back
#define all(x)      x.begin(),x.end()
#define memset(a,b) memset(a,b,sizeof(a));
#define IOS         ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
void file()
{
#ifdef ONLINE_JUDGE
#else
    freopen("D:/LSNU/codeforces/duipai/data.txt","r",stdin);
      freopen("D:/LSNU/codeforces/duipai/WA.txt","w",stdout);
#endif
}
const int N=1e3+5;
vector<pair<int,int> >G[N];
int dis[N],dp[N];
struct node
{
    int u,v,dis;
};
void dijkstra(int s)
{
    memset(dis,0x3f);
    dis[s]=0;
    priority_queue<pair<int,int> >pri;
    pri.push({0,s});
    while(!pri.empty())
    {
        pair<int,int> u=pri.top();
        pri.pop();
        if(dis[u.S]<-u.F)
            continue;
        for(auto it:G[u.S])
        {
            if(dis[it.F]>dis[u.S]+it.S)
            {
                dis[it.F]=dis[u.S]+it.S;
                pri.push({-dis[it.F],it.F});
            }
        }
    }
}
int dfs(int u)
{
    if(dp[u])
        return dp[u];
    if(u==2)
        return 1;
    for(auto it:G[u])
    {
        if(dis[u]>dis[it.F])
            dp[u]+=dfs(it.F);
    }
    return dp[u];
}
signed main()
{
    IOS;
   // file();
    int n,m;
    while(cin>>n&&n)
    {
        for(int i=1;i<=n;i++)
            G[i].clear(),dp[i]=0;
        cin>>m;
        vector<node>vec(m);
        for(auto &it:vec)
        {
            cin>>it.u>>it.v>>it.dis;
            G[it.u].pb({it.v,it.dis});
            G[it.v].pb({it.u,it.dis});
        }
        dijkstra(2);
        cout<<dfs(1)<<endl;
    }
    return 0;
}

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