POJ 3463 Sightseeing（次短路）转自'wind

大意思路转自http://hi.baidu.com/ofeitian/item/bf283031b29e86f02784f451

题目大意：求源点S到终点T的最短路的数量和比最短路长1的数量。

解题思路：我们可以利用dijstra算法的思想，只需在其中进行一些改进即可。可以先定义一个二维的数组dist[N][2]。dist[i][0]代表源点S到点i的最短路，dist[i][1]代表源点S到点i的次短路。初始化dist[S][0]=0，其余的都初始化为无穷大。然后定义一个二维数组waynum[N][2]记录路径方法数，waynum[S][0]=1，其余初始为0。再定义一个标记数组vis[N][2]，初始vis[S][0]被标记已访问，其余未访问。采用dijstra算法的思想，每次从dist[N][2]中选择一个未被标记且值最小的点dist[v][flag]（可能这个值是这个点的最短路，也可能是次短路，只有当此点的最短路被标记了次才可能选中此点的次短路）。再用这个值val去更新此点的邻接点u。更新的方法如下：

（1）如果val小于dist[u][0]，则dist[u][1]=dist[u][0],waynum[u][1]=waynum[u][0],dist[u][0]=val.waynum[u][0]=waynum[v][flag]。否则转（2）

（2）如果val 等于dist[u][0],则waynum[u][0] += waynum[v][flag]; 否则转（3）

（3）如果val小于dist[u][1],则dist[u][1]=val.waynum[u][1]=waynum[v][flag]。

否则转（4）

（4）如果val等于dist[u][1]，则waynum[u][1] +=waynum[v][flag].否则什么都不做。

这样循环计算2*n-1次就可以计算出源点到所有点的最短路和次短路的方法数了，而对于终点T，如果次短路比最短路大1则答案为最短路和次短路的方法数之和，否则就为最短路的方法数。

代码转自'wind

分析： 在原来的dijkstra()上加一点变动，d[i][0]记录最短路的长度，d[i][1]记录次短路的长度，dp[i][0] 到达当前点最短路的条数，dp[i][1]到达当前点次短路的条数。

View Code 

/************************************************************************************
  采用dijstra算法的思想，每次从dis[N][2]中选择一个未被标记且值最小的点dis[v][p]
  (可能这个值是这个点的最短路，也可能是次短路，只有当此点的最短路被标记了次才可能选中
  此点的次短路).再用这个值val去更新此点的邻接点u.更新的方法如下:
  ①如果val小于dis[u][0]
    dis[u][1]=dis[u][0]
    waynum[u][1]=waynum[u][0]
    dis[u][0]=val
    waynum[u][0]=waynum[v][p]
    否则转②
  ②如果val=dis[u][0]
    waynum[u][0]+=waynum[v][p] 
    否则转③
  ③如果val小于dis[u][1], 
    dis[u][1]=val
    waynum[u][1]=waynum[v][p]
    否则转④
  ④如果val等于dis[u][1]
    waynum[u][1]+=waynum[v][p]
************************************************************************************/
#include<stdio.h>
#include<string.h>
#define INF    999999999
#define clr(x)memset(x,0,sizeof(x))
const int maxn=1005;
struct node
{
    int to,next,w;
}e[100005];
int tot;
int head[maxn];
void add(int s,int u,int wi)
{
    e[tot].w=wi;
    e[tot].to=u;
    e[tot].next=head[s];
    head[s]=tot++;
}
int n,m,st,en;
int d[maxn][2];
int dp[maxn][2];
int v[maxn][2];
int dijkstra()
{
    int i,j,u,tmp,p,k,dis;
    clr(v);
    clr(dp);
    memset(dp,0,sizeof(dp));
    for(i=1;i<=n;i++)
        d[i][0]=d[i][1]=INF;
    d[st][0]=0;
    dp[st][0]=1;
    for(i=1;i<n*2;i++)
    {
        tmp=INF;
        for(j=1;j<=n;j++)
        {
            if(!v[j][0]&&d[j][0]<tmp)
            {
                tmp=d[j][0];
                u=j;
                p=0;
            }
            else if(!v[j][1]&&d[j][1]<tmp)
            {
                tmp=d[j][1];
                u=j;
                p=1;
            }
        }
        if(tmp==INF)
            break;
        v[u][p]=1;
        for(j=head[u];j;j=e[j].next)
        {
            k=e[j].to;
            dis=e[j].w;
            if(tmp+dis<d[k][0])
            {
                d[k][1]=d[k][0];
                d[k][0]=tmp+dis;
                dp[k][1]=dp[k][0];
                dp[k][0]=dp[u][p];
            }
            else if(tmp+dis==d[k][0])
                dp[k][0]+=dp[u][p];
            else if(tmp+dis<d[k][1])
            {
                d[k][1]=tmp+dis;
                dp[k][1]=dp[u][p];
            }
            else if(tmp+dis==d[k][1])
                dp[k][1]+=dp[u][p];
        }
    }
    if(d[en][0]+1==d[en][1])
        return dp[en][0]+dp[en][1];
    return dp[en][0];
}
int main()
{
    int a,b,c,t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        clr(head);
        tot=1;
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        scanf("%d%d",&st,&en);
        printf("%d\n",dijkstra());
    }
    return 0;
}