hdu 1142 记忆化搜索

题目是这样的，貌似一开始我这个英语搓的人还理解错了。。。orz

http://acm.hdu.edu.cn/showproblem.php?pid=1142

就是最短路，只不过用dijkstra是从终点到源点，然后再dfs从源点开始搜。。。

好吧，这个记忆化搜索还是挺好用的。。。

#include<iostream>
#include<cstring>
const int N=1010;
const int inf=99999999;
using namespace std;

int edge[N][N];
int dist[N];
int visited[N],s[N];
int n,m;

void Dijkstra(int v0){
    memset(visited,0,sizeof(visited));
    for(int i=1;i<=n;i++){
        dist[i]=edge[v0][i];
    }
    dist[v0]=0;
    visited[v0]=1;
    for(int i=1;i<n;i++){
        int min=inf,u=v0;
        for(int j=1;j<=n;j++){
            if(!visited[j]&&dist[j]<min)
                u=j,min=dist[j];
        }
        if(min==inf)return ;
        visited[u]=1;
        for(int k=1;k<=n;k++){
            if(!visited[k]&&edge[u][k]<inf&&dist[u]+edge[u][k]<dist[k]){
                dist[k]=edge[u][k]+dist[u];
            }
        }
    }
}
//记忆化搜索
int dfs(int v){
    if(s[v]) return s[v];  //如果该点已经访问过了，就返回到该点的路径数
    if(v==2)return 1; //找到终点，返回1条路
    for(int i=1;i<=n;i++){
        if(edge[v][i]<inf&&dist[i]<dist[v]){
            s[v]+=dfs(i);
        }
    }
    return s[v]; //返回到该点的所有路径数
}

int main(){
    while(scanf("%d",&n)!=EOF){
        if(n==0)break;
        scanf("%d",&m);
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                if(i==j)edge[i][j]=0;
                else edge[i][j]=inf;
            }
        }
        int a,b,dis;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&dis);
            if(edge[a][b]>=dis){
                edge[a][b]=edge[b][a]=dis;
            }
        }
        Dijkstra(2);
        memset(s,0,sizeof(s));
        int count=dfs(1);
        printf("%d\n",count);
    }
    return 0;
}