HDU-3191 How Many Paths Are There 次最短路

#include "stdio.h"
#include "string.h"
#include "vector"
#include "queue"
using namespace std;
const int maxn = 210;
const int inf = 1<<30;
int n,m,s,e;
int dis[maxn][2],cnt[maxn][2],vis[maxn][2];
//dis[i][0]表示到达点i最短路的长度,dis[i][1]表示次短路的长度
//cnt[i][0]表示到达点i最短路的条数,cnt[i][1]表示次短路的条数
struct node
{
int to,w;
node( int a,int b ) { to = a; w = b; }
};
vector<node>map[maxn];

struct Node
{
int v,dis;
int mark;	//标记最短还是次短
bool operator < ( const Node &p ) const
{
if( p.dis != dis )
return p.dis < dis;

return p.v < v;//这儿如果不按顶点的大小排序，就wa了。
}
};

void Dijstra( int s,int e )
{
for( int i = 0; i < n; i ++ )
{
dis[i][0] = dis[i][1] = inf;
cnt[i][0] = cnt[i][1] = 0;
}
dis[s][0] = 0;  cnt[s][0] = 1;
memset( vis,0,sizeof(vis) );
priority_queue<Node>que;
Node u,v;
u.dis = 0; u.mark = 0; u.v = s;
que.push( u );
while( !que.empty())
{
u = que.top();  que.pop();
if( vis[u.v][u.mark] )
continue;
vis[u.v][u.mark] = true;
//松驰时有四种情况
for( int i = 0;i < map[u.v].size(); i ++ )
{
node x = map[u.v][i];
if( !vis[x.to][0] && dis[x.to][0] > u.dis + x.w )  //找到一个更短的距离，则把原来最短的距离作为次短的距离，同时更新最短的.
{
if( dis[x.to][0] != inf )					   //可能为次短路
{
v.v = x.to; v.dis = dis[x.to][0]; v.mark = 1;
dis[x.to][1] = dis[x.to][0];
cnt[x.to][1] = cnt[x.to][0];
que.push( v );
}
dis[x.to][0] = u.dis + x.w;
cnt[x.to][0] = cnt[u.v][u.mark];
v.v = x.to; v.dis = dis[x.to][0]; v.mark = 0;
que.push( v );
}
else if( !vis[x.to][0] && dis[x.to][0] == u.dis + x.w )
cnt[x.to][0] += cnt[u.v][u.mark];

else if( !vis[x.to][1] && dis[x.to][1] > u.dis + x.w )//不可以更新最短距离，但可以更新次短的
{
dis[x.to][1] = u.dis + x.w;
cnt[x.to][1] = cnt[u.v][u.mark];
v.v = x.to; v.dis = dis[x.to][1]; v.mark = 1;
que.push( v );
}
else if( !vis[x.to][1] && dis[x.to][1] == u.dis + x.w )//找到一条新的相同距离的次短路
cnt[x.to][1] += cnt[u.v][u.mark];
}
}
}

int main()
{
int a,b,c;
//freopen( "data.txt","r",stdin );
while( scanf("%d%d%d%d",&n,&m,&s,&e) == 4 )
{
for( int i = 0; i < n; i ++ )
map[i].clear();
for( int i = 1; i <= m; i ++ )
{
scanf("%d%d%d",&a,&b,&c);
map[a].push_back( node(b,c) );
}
Dijstra( s,e );
printf("%d %d\n",dis[e][1],cnt[e][1]);
}
return 0;
}

#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 210;
const int inf = 1<<29;
int n,m,s,e;
int dis[maxn],outque[maxn],vis[maxn];
struct node
{
int id,w;
node( int a,int b ) { id = a; w = b; }
};
struct Node
{
int to;
int g,f;
bool operator < ( const Node &r ) const
{
if( r.f == f )    return r.g < g;
return r.f < f;
}
};
vector<node>map[maxn],maps[maxn];

bool SPEA( int s )
{
queue<int>que;
memset( outque,0,sizeof(outque) );
memset( vis,0,sizeof(vis) );
for( int i = 0; i < n; i ++ )
dis[i] = inf;
dis[s] = 0;
que.push( s );
vis[s] = true;
while( !que.empty() )
{
int x = que.front(); que.pop();
vis[x] = false;
outque[x] ++;
if( outque[x] > n )
return false;
for( int i = 0; i < map[x].size(); i ++ )
{
node u = map[x][i];
if( u.w != inf && dis[u.id] > dis[x] + u.w )
{
dis[u.id] = dis[x] + u.w;
if( !vis[u.id] )
{
vis[u.id] = true;
que.push( u.id );
}
}
}
}
return true;
}

int Astar( int start,int end,int k )
{
Node e,u;
int cnt = 0;        //标记终点出队次数
priority_queue<Node>que;
if( start == end )
k ++;
if( dis[start] == inf )
return -1;
e.to = start;   e.g = 0;
e.f = e.g + dis[e.to];
que.push(e);
while( !que.empty() )
{
e = que.top(); que.pop();
if( e.to == end )
cnt ++;
if( cnt == k )
{
return e.g;
}
for( int i = 0; i < maps[e.to].size(); i ++ )
{
node v = maps[e.to][i];
u.to = v.id;
u.g = e.g + v.w;
u.f = u.g + dis[u.to];
que.push(u);
}
}
return -1;
}

int main()
{
freopen("data.txt","r",stdin);
int a,b,c;
while( scanf("%d%d%d%d",&n,&m,&s,&e) == 4 )
{
for( int i = 1; i <= m; i ++ )
{
scanf("%d%d%d",&a,&b,&c);
maps[a].push_back( node(b,c) );
map[b].push_back( node(a,c) );
}
SPEA( e );
int ans = Astar( s,e,2 );
printf("%d\n",ans);
}
return 0;
}