hdu-1142（记忆化搜索+dij）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1142

思路：1、不是求最短路径，而是求如果两个点A，B直接相连，且A到终点的距离大于B到终点的距离，求这样A，B之间的通路的个数。

2、以终点为起点来进行dij

3、记忆化搜索是，用p[i]数组记录符合条件的路径数量。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = ;
const int INF = ;
int n,m,vis[maxn],dis[maxn],e[maxn][maxn],p[maxn];
void Init()
{
    int i,j;
    memset(vis,,sizeof(vis));
    memset(p,,sizeof(p));
    for(i=;i<=n;i++)
       for(j=;j<=n;j++)
        {
               if(i==j) e[i][j]=;
               else e[i][j]=INF;
        }
}
void Dij()
{
    int pos,mx,i,j;
    for(i=;i<=n;i++) dis[i]=e[][i];
    dis[]=;
    vis[]=;
    for(i=;i<n;i++)
    {
        mx=INF,pos=-;
        for(j=;j<=n;j++)
        {
            if(vis[j]==&&mx>dis[j])
            {
                mx=dis[j];
                pos=j;
            }
        }
        if(pos<) break;
        vis[pos]=;
        for(j=;j<=n;j++)
        {
            if(vis[j]==&&dis[j]>dis[pos]+e[pos][j]&&e[pos][j]!=INF)
            {
                dis[j]=dis[pos]+e[pos][j];
            }
        }
    }
}
int Dfs(int s)
{
    int i,sum=;
    if(p[s]) return p[s];
    if(s==) return ;
    for(i=;i<=n;i++)
    {
        if(e[s][i]!=INF&&dis[s]>dis[i])
        {
            if(p[i]) sum+=p[i];
            else sum+=Dfs(i);
        }
    }
    sum+=p[s];
    p[s]=sum;
    return sum;
}
int main(void)
{
    int i,j,x,y,z;
    while(~scanf("%d",&n)&&n)
    {
        scanf("%d",&m);
        Init();
        for(i=;i<m;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            e[x][y]=e[y][x]=z;
        }
        Dij();
        printf("%d\n",Dfs());
    }
    return ;
}