G - Walk Through the Forest (UVA - 10917)
阿新 • • 發佈:2018-02-15
道路 names urn break blog pri res bsp bre
- 題目大意
一個人,他只會沿著如下條件的道路(A,B)走:存在一條從B出發回家的路徑,比所有從A出發回家的路徑都要短。我們的任務是要找出一共有有多少條不同的回家路徑。
- 解題思路
先用dijkstra預處理出終點到每個點的最短路,然後將滿足行走條件的A、B(除行走條件外,還要滿足一個前提,即A、B之間要有邊)用一條有向邊連起來(A->B),然後利用記憶化搜索來解決這個問題。
- 代碼
#include<cstdio> #include<cstring> #include<queue> using namespace std; const int M=1e5; const int INF=0x3f3f3f; int d[1050],vis[1050],g[1050][1050],maps[1050]; void init(int n) { for(int i=1;i<=n;i++) { for(int j=1;j<=n;j++) { g[i][j]=INF; } } } void dijkstra(int s, int n) { memset(d, INF, sizeof(d)); memset(vis, 0, sizeof(vis)); d[s] = 0; while(1) { int v = -1; for(int u = 1; u <= n; u++) { if(!vis[u] && (v == -1 || d[u] < d[v])) v = u; } if(v == -1) break; vis[v] = 1; for(int u = 1; u <= n; u++) { if(!vis[u] && d[u] > d[v] + g[v][u]) { d[u] = d[v] + g[v][u]; } } } } int dfs(int i,int n) { int sum=0; if(i==2) return 1; if(~maps[i]) return maps[i]; for(int j=1;j<=n;j++) { if(g[j][i]<INF&&d[i]>d[j]) sum+=dfs(j,n); } maps[i]=sum; return maps[i]; } int main() { int n,m,a,b,c; while(scanf("%d",&n)&&n!=0) { scanf("%d",&m); init(n); for(int i=1;i<=m;i++) { scanf("%d%d%d",&a,&b,&c); g[a][b]=c; g[b][a]=c; } memset(maps,-1,sizeof(maps)); dijkstra(2,n); printf("%d\n",dfs(1,n)); } return 0; }
G - Walk Through the Forest (UVA - 10917)