poj 3463 Sightseeing

题目：Sightseeing

题意：给出一张有向图，求从起点到终点的最短路条数和比最短路长1的路径条数和。

思路：
dijkstra求出最短路和次短路以及条数。如果次短路只比最短路长1，输出条数和。否则仅输出最短路的条数。
参考这位dalao的代码。

代码：#include <cstdio>
#include <iostream>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <set>
#include <cstring>
#include <map>
#include <cmath>
using namespace std;

#define maxn 10000

int n,m;
int s,e;

struct Edge {
int x,y,c;
Edge(int xx=0,int yy=0,int cc=0) {
x=xx,y=yy,c=cc;
}
bool operator < (const Edge& oth) const {
return y>oth.y||(y==oth.y&&x<oth.x);
}
};

vector<Edge> a[maxn+5];

int dist1[maxn+5],dist2[maxn+5]; //最、次短路
int cnt[maxn+5][3]; //最、次短路的条数

bool c[maxn+5][3];
priority_queue<Edge> p;

void init() { //初始化
for(int i=1; i<=maxn; i++) {
a[i].clear();
dist1[i]=dist2[i]=(1<<29);
}
memset(c,0,sizeof(c));
memset(cnt,0,sizeof(cnt));
priority_queue<Edge> emp;
p=emp;
}

void dijkstra() { //dijkstra求最短路和次短路
dist1[s]=0,cnt[s][1]=1;
p.push(Edge(s,0,1));

while(!p.empty()) {
Edge x=p.top();
p.pop();
if(c[x.x][x.c]) continue;
c[x.x][x.c]=true;

for(int i=0; i<a[x.x].size(); i++) {
if(dist1[a[x.x][i].x]>x.y+a[x.x][i].y) { //更新最短路
dist2[a[x.x][i].x]=dist1[a[x.x][i].x];
cnt[a[x.x][i].x][2]=cnt[a[x.x][i].x][1];
dist1[a[x.x][i].x]=x.y+a[x.x][i].y;
cnt[a[x.x][i].x][1]=cnt[x.x][x.c];
p.push(Edge(a[x.x][i].x,dist1[a[x.x][i].x],1));
p.push(Edge(a[x.x][i].x,dist2[a[x.x][i].x],2)); //注意！由于这里也更新了次短路，所以不能忘记这一行
} else if(dist1[a[x.x][i].x]==x.y+a[x.x][i].y) { //更新最短路的条数
cnt[a[x.x][i].x][1]+=cnt[x.x][x.c];
} else if(dist2[a[x.x][i].x]>x.y+a[x.x][i].y) { //更新次短路
dist2[a[x.x][i].x]=x.y+a[x.x][i].y;
cnt[a[x.x][i].x][2]=cnt[x.x][x.c];
p.push(Edge(a[x.x][i].x,dist2[a[x.x][i].x],2));
} else if(dist2[a[x.x][i].x]==x.y+a[x.x][i].y) { //更新次短路的条数
cnt[a[x.x][i].x][2]+=cnt[x.x][x.c];
}
}
}
}

void readin() { //读入
scanf("%d%d",&n,&m);
for(int i=1; i<=m; i++) {
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
a[x].push_back(Edge(y,z));
}
scanf("%d%d",&s,&e);
}

void print() { //输出asd
int ans=cnt[e][1];
if(dist1[e]+1==dist2[e]) ans+=cnt[e][2]; //仅当最短路只比次短路小1时才有效
printf("%d\n",ans);
}

int main() {
int T;
scanf("%d",&T);

while(T--) {
init();
readin();
dijkstra();
print();
}
return 0;
}