poj 2662 A Walk Through the Forest

题目意思，可以从A走到B当且仅当B到终点的最短路径小于A到终点的最短路径

问一共存在几条这样的路径

我用dijkstra各做了一遍，呵呵

个中细节还需细细体味啊

View Code

#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
const int INF = 9999999;
struct NODE{
    int v,w;
    int next;
}list[10000000];
int tot;
int head[1010];
int dis[1010];
int vis[1010];
int sum[1010];
queue<int> Q;
int n,m;
void init()
{
    int i;
    for(i=0;i<=1000000;i++)
        list[i].w=INF;
    memset(head,-1,sizeof(head));
}
void add(int s,int t,int w)
{
    list[tot].v=t;
    list[tot].w=w;
    list[tot].next=head[s];
    head[s]=tot++;
}
void spfa(int src)
{
    int i,j,u;
    while(!Q.empty())
        Q.pop();
    memset(vis,0,sizeof(vis));
    for(i=0;i<=1000;i++)
        dis[i]=INF;
    Q.push(src);
    vis[src]=1;
    dis[src]=0;
    while(!Q.empty())
    {
        u=Q.front();
        Q.pop();
        vis[u]=0;
        for(j=head[u];j!=-1;j=list[j].next)
        {
            int e=list[j].v;
            if(dis[e]>list[j].w+dis[u])
            {
                dis[e]=list[j].w+dis[u];
                if(!vis[e])
                {
                    Q.push(e);
                    vis[e]=1;
                }
            }
        }
    }
}
int dfs(int i)
{
    int j;
    if(i==2) return 1;
    if(sum[i]!=-1) return sum[i];//很重要，避免重复搜索，否则超时。
    int cnt=0;
//    printf("i disi=%d %d\n",i,dis[i]);
    for(j=head[i];j!=-1;j=list[j].next)
    {
        if(dis[i]>dis[list[j].v]&&list[j].w<INF)
        {    //printf("i=%d to=%d\n",i,list[j].v);//这里输出后，dfs的调用过程就一清二楚了
            cnt+=dfs(list[j].v); 
        //    printf("cnt=%d\n",cnt);
        }
    }
   sum[i]=cnt;//printf("sum[%d]=%d\n",i,sum[i]);
    return sum[i];
}
int main()
{
    int a,b,d;
    while(scanf("%d",&n),n)
    {
        scanf("%d",&m);
        init();
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&d);
            add(a,b,d);
            add(b,a,d);
        }
        spfa(2);
        memset(sum,-1,sizeof(sum));
        int ans=dfs(1);
        printf("%d\n",ans);
    }
    return 0;
}

/*
5 6
1 3 2
1 4 2
3 4 3
1 5 12
4 2 34
5 2 24
sum[5]=1
sum[4]=1
sum[1]=2
2
7 8
1 3 1
1 4 1
3 7 1
7 4 1
7 5 1
6 7 1
5 2 1
6 2 1
sum[6]=1
sum[5]=1
sum[7]=2
sum[4]=2
sum[3]=2
sum[1]=4
4
*/