Uva 10917 Walk Through the Forest（DAG拓扑排序）

题意：jimmy下班需要穿过一个森林，所以打算每天沿着一条不同的路径回家，他只沿着满足如下条件的(A,B)走：存在一条从B出发的回家路，比所有从A出发的回家路都短，问你

一共有多少种回家方案，1为家，2为起点。

分析：求一遍从2开始的单元最短路，然后从新构图，只保留d(a) > d(b) 的 (b,a)边，注意这一定是一个DAG(有向无环图)，然后拓扑排序后DP。

#include <cstdio>
#include <stack>
#include <queue>
#include <vector>
#include <cstdio>
#include <utility>
#include <cstring>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
long long n,m,f[1005],d[1005],rd[1005];
bool jud[1005];
typedef pair <int,int> Pii;
vector <Pii> G[1005],G2[1005];
deque <int> q;
void spfa()
{
memset(d,INF,sizeof(d));
memset(jud,0,sizeof(jud));
d[2] = 0;
q.push_back(2);
while(!q.empty())
{
int u = q.front();
for(Pii x : G[u])
{
int v = x.first,val = x.second;
if(d[u] + val < d[v])
{
d[v] = d[u] + val;
if(!jud[v])
{
jud[v] = true;
q.push_back(v);
}
}
}
jud[u] = false;
q.pop_front();
}
}
int main()
{
while(scanf("%d",&n) && n)
{
scanf("%d",&m);
memset(f,0,sizeof(f));
memset(rd,0,sizeof(rd));
for(int i = 1;i <= n;i++)
{
G[i].clear();
G2[i].clear();
}
for(int i = 1;i <= m;i++)
{
int x,y,v;
scanf("%d%d%d",&x,&y,&v);
G[x].push_back(make_pair(y,v));
G[y].push_back(make_pair(x,v));
}
spfa();
for(int u = 1;u <= n;u++)
for(Pii x : G[u])
{
int v = x.first;
if(d[u] < d[v])
{
G2[u].push_back(x);
rd[v]++;
}
}
f[2] = 1;
q.push_back(2);
while(!q.empty())
{
int u = q.front();
for(Pii x : G2[u])
{
int v = x.first;
f[v] += f[u];
rd[v]--;
if(!rd[v]) q.push_back(v);
}
q.pop_front();
}
cout<<f[1]<<endl;
}
}