poj 2662 A Walk Through the Forest

题目意思,可以从A走到B当且仅当B到终点的最短路径小于A到终点的最短路径

问一共存在几条这样的路径

我用dijkstra各做了一遍,呵呵

个中细节还需细细体味啊

View Code
#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
const int INF = 9999999;
struct NODE{
int v,w;
int next;
}list[10000000];
int tot;
int head[1010];
int dis[1010];
int vis[1010];
int sum[1010];
queue<int> Q;
int n,m;
void init()
{
int i;
for(i=0;i<=1000000;i++)
list[i].w=INF;
memset(head,-1,sizeof(head));
}
void add(int s,int t,int w)
{
list[tot].v=t;
list[tot].w=w;
list[tot].next=head[s];
head[s]=tot++;
}
void spfa(int src)
{
int i,j,u;
while(!Q.empty())
Q.pop();
memset(vis,0,sizeof(vis));
for(i=0;i<=1000;i++)
dis[i]=INF;
Q.push(src);
vis[src]=1;
dis[src]=0;
while(!Q.empty())
{
u=Q.front();
Q.pop();
vis[u]=0;
for(j=head[u];j!=-1;j=list[j].next)
{
int e=list[j].v;
if(dis[e]>list[j].w+dis[u])
{
dis[e]=list[j].w+dis[u];
if(!vis[e])
{
Q.push(e);
vis[e]=1;
}
}
}
}
}
int dfs(int i)
{
int j;
if(i==2) return 1;
if(sum[i]!=-1) return sum[i];//很重要,避免重复搜索,否则超时。
int cnt=0;
// printf("i disi=%d %d\n",i,dis[i]);
for(j=head[i];j!=-1;j=list[j].next)
{
if(dis[i]>dis[list[j].v]&&list[j].w<INF)
{ //printf("i=%d to=%d\n",i,list[j].v);//这里输出后,dfs的调用过程就一清二楚了
cnt+=dfs(list[j].v);
// printf("cnt=%d\n",cnt);
}
}
sum[i]=cnt;//printf("sum[%d]=%d\n",i,sum[i]);
return sum[i];
}
int main()
{
int a,b,d;
while(scanf("%d",&n),n)
{
scanf("%d",&m);
init();
while(m--)
{
scanf("%d%d%d",&a,&b,&d);
add(a,b,d);
add(b,a,d);
}
spfa(2);
memset(sum,-1,sizeof(sum));
int ans=dfs(1);
printf("%d\n",ans);
}
return 0;
}

/*
5 6
1 3 2
1 4 2
3 4 3
1 5 12
4 2 34
5 2 24
sum[5]=1
sum[4]=1
sum[1]=2
2
7 8
1 3 1
1 4 1
3 7 1
7 4 1
7 5 1
6 7 1
5 2 1
6 2 1
sum[6]=1
sum[5]=1
sum[7]=2
sum[4]=2
sum[3]=2
sum[1]=4
4
*/

posted @ 2011-09-26 15:19  Because Of You  Views(273)  Comments(0Edit  收藏  举报