A Walk Through the Forest HDU - 1142 (Dijkstra+DFS)

讀錯題了...要的是滿足:如果從a到b,a到終點的距離>b到終點的距離,這個就算有效的

問的是以上有效的有幾個,而不是多少從起點到終點的路徑...

TLE了還以爲是自己沒抄板子dijkstra寫錯了...

#include <iostream>
#include <cstdio>
#include <cstring>
#include<algorithm>
#include<sstream>
#include<queue>
#define ull unsigned long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 1003;
int n,m;
struct EDGE
{
    int to,dis;
    EDGE()
    {
        to = dis = 0;
    }
    EDGE(int a,int b)
    {
        to = a;
        dis = b;
    }
};
vector<EDGE> vec[1003];
int dis[1003];
bool vis[1003][1003];
int path[1003];
void init()
{
    for(int i=0; i<=n; i++)
    {
        vec[i].clear();
        dis[i] = INF;
    }
    memset(vis,false,sizeof(vis));
}
void dijkstra(int start,int endd)
{
    priority_queue<pair<int,int> > que;
    que.push(make_pair(0,endd));
    dis[endd] = 0;
    EDGE a;
    while(!que.empty())
    {
        pair<int,int> P = que.top();
        que.pop();
        int pos = P.second;
        int len = vec[pos].size();
        for(int i=0; i<len; i++)
        {
            a = vec[pos][i];
            if(dis[a.to] > dis[pos] + a.dis)
            {
                dis[a.to] = dis[pos] + a.dis;
                que.push(make_pair(dis[a.to],a.to));
            }
        }
    }
}
int DFS(int src)
{
    if(path[src] != -1)
        return path[src];
    if(src == 2)
        return 1;
    path[src] = 0;
    for(int i=1;i<=n;i++)
    {
        if(dis[i] < dis[src] && vis[i][src])
        {
            path[src] += DFS(i);
        }
    }
    return path[src];
}
int main()
{
//    freopen("in.txt","r",stdin);

    ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    while(cin>>n>>m && m+n)
    {
        init();
        int a,b,c;
        for(int i=0; i<m; i++)
        {
            cin>>a>>b>>c;
            vec[a].push_back(EDGE(b,c));
            vec[b].push_back(EDGE(a,c));
            vis[a][b] = vis[b][a] = true;
        }
        dijkstra(1,2);
        memset(path,-1,sizeof(path));
//        for(int i=1;i<=n;i++)
//        {
//            cout<<i<<"  "<<dis[i]<<endl;
//        }
        cout<<DFS(1)<<'\n';
    }
    return 0;
}