统计单源最短路径的条数 （DFS+Dijstra算法）和记忆化搜索

主要参考了http://blog.csdn.net/qinaide_lixiaoshuo/article/details/7880264

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#define inf 0x3fffffff

int N,M;//N represents the number of points,and M stands for the number of edges;

int map[1024][1024];

int cnt;

int ans;

int dis[1024];

int visit[1024];

const int sta=1;

const int end=2;

struct Edge{
	int v;
	int val;
	int next;
}e[1111111];

int head[1024];

int idx;

int dijkstra()
{
	memset(visit,0,sizeof(visit));
	for(int i=1;i<=N;i++)
		dis[i]=inf;
	dis[sta]=0;
	for(int j=1;j<=N;j++)
	{
		int t=inf;
		int pos;
		for(int i=1;i<=N;i++)
		{
			if(!visit[i]&&t>dis[i])
			{
				t=dis[i];
				pos=i;
			}
		}
		visit[pos]=1;
		for(i=1;i<=N;i++)
		{
			if(!visit[i]&&dis[i]>dis[pos]+map[pos][i]&&map[pos][i]!=0x3f3f3f3f)
			{
				dis[i]=dis[pos]+map[pos][i];
			}
		}
	}
	if(dis[end]==inf)
		return -1;
	else
		return dis[end];
}

void addedge(int a,int b,int val)
{
	e[idx].v=b;
	e[idx].val=val;
	e[idx].next=head[a];
	head[a]=idx++;
}

void DFS(int index,int val)
{
//	printf("%d___%d___\n",index,val);
	if(index==end)
	{
		if(val==ans)
		{
			cnt++;
			return;
		}
		else
			return;
	}
	if(val>=ans)
		return;
	else
	{
		for(int p=head[index];p!=-1;p=e[p].next)
		{
			int v=e[p].v;
			if(!visit[v])
			{
				visit[v]=1;
				DFS(v,val+e[p].val);
				visit[v]=0;
			}
		}
	}
}





int main()
{
	int aa,bb,cc;
	while(scanf("%d",&N)!=EOF,N)
	{
		memset(head,-1,sizeof(head));
		memset(map,0x3f,sizeof(map));
		scanf("%d",&M);
		int t=inf;
		idx=0;
		for(int i=1;i<=M;i++)
		{
			scanf("%d%d%d",&aa,&bb,&cc);
			if(map[aa][bb]>cc)//判断重边
			{
				map[aa][bb]=cc;
				map[bb][aa]=cc;
				addedge(aa,bb,cc);
				addedge(bb,aa,cc);
			}
		}
		ans=dijkstra();
	//	printf("%d____\n",ans);
		if(ans==-1)
			printf("0\n");
		else
		{
			cnt=0;
			memset(visit,0,sizeof(visit));
			visit[sta]=1;
			DFS(sta,0);
			printf("%d\n",cnt);
		}
	}
	return 0;
}

 

上面写的有些啰嗦：

下面是整理过的：

#define inf 0x3fffffff
int N,M;//顶点和边的数目
int map[1024][1024];
int cnt;//最短路径的条数
int ans;//最短路径
int dis[1024];//最短路径
int visit[1024];
const int sta=0;
const int end=2;
int dijkstra()
{
	memset(visit,0,sizeof(visit));
	for(int i=0;i<N;i++)//不使用dis[0]
		dis[i]=inf;
	dis[sta]=0;
	visit[sta]=1;
	for(int j=1;j<N;j++)
	{
		int t=inf;
		int pos=sta;
		for(int i=0;i<N;i++)
		{
			if(!visit[i]&&t>dis[i])
			{
				t=dis[i];
				pos=i;
			}
		}
		visit[pos]=1;
		for(int i=0;i<N;i++)
		{
			if(!visit[i]&&dis[i]>dis[pos]+map[pos][i]&&map[pos][i]!=inf)
			{
				dis[i]=dis[pos]+map[pos][i];
			}
		}
	}
	if(dis[end]==inf)
		return -1;
	else
		return dis[end];
}
//index开始的位置，val 当前的路径权值和 当val<ans返回
void DFS(int index,int val,stack<int> s)
{
	if(index==end)
	{
		if(val==ans)
		{
			cnt++;
			while(!s.empty())
			{
				cout<<s.top()<<" ";
				s.pop();
			}
			cout<<sta<<" ";
			cout<<endl;
			return;
		}
		else
			return;
	}
	if(val>=ans)
		return;
	else
	{
		int p=map[index][index+1];
		for(;p!=-1&&p<N;p++)
		{
			int v=p;
			if(!visit[v])
			{
				visit[v]=1;
				s.push(v);
				DFS(v,val+map[index][p],s);
				s.pop();
				visit[v]=0;
			}
		}
	}
}
int main()
{
	int aa,bb,cc;
	while(scanf("%d",&N)!=EOF,N)
	{
	//	memset(head,-1,sizeof(head));//初始化head
		memset(map,0x3f,sizeof(map));
		scanf("%d",&M);
		int t=inf;
		for(int i=0;i<M;i++)
		{
			scanf("%d%d%d",&aa,&bb,&cc);
			if(map[aa][bb]>cc)
			{
				map[aa][bb]=cc;
				map[bb][aa]=cc;
			}
		}
		ans=dijkstra();
		cout<<"最短路径为："<<ans<<" "<<endl;
		if(ans!=-1)
		{
			cnt=0;
			memset(visit,0,sizeof(visit));
			stack<int> s;
			DFS(sta,0,s);
			cout<<"条数"<<cnt<<endl;
		}
	}
}

 

 

 之后使用的是记忆化搜索统计最短路径的条数:

int m,n,map[1024][1024],des[1024],dis[1024],dep[1024];
int inf=0x7fffffff;
int Dijkstra(int p)
{
	dis[p]=0;
	for(int i=0;i<n;i++)
	{
		int min=inf,pos=0,t;
		for(int j=0;j<n;j++)
		{
			if(!des[j]&&dis[j]<min)
			{
				min=dis[j];
				pos=j;
			}
		}
		des[pos]=1;
		for(int j=0;j<n;j++)
		{
			if(((!des[j])&&(map[pos][j]+dis[pos]<dis[j]))&&map[pos][j]!=inf)
				dis[j]=map[pos][j]+dis[pos];
		}
	}
	return 0;
}
int DFS(int x)
{
	if(dep[x]!=0)
	{
		return dep[x];
	}
	if(x==0)
		return 1;
	for(int i=0;i<n;i++)
	{
		if(dis[i]<dis[x]&&map[x][i]!=inf)
		{
			dep[x]+=DFS(i);
		}
	}
	return dep[x];
}

int main()
{
	while(scanf("%d",&n),n)
	{
		scanf("%d",&m);
		for(int i=0;i<n;i++)
		{
			for(int j=0;j<n;j++)
			{
				map[i][j]=inf;
			}
			des[i]=0;
			dis[i]=inf;
			dep[i]=0;
		}
		for(int i=0;i<m;i++)
		{
			int x,y,v;
			scanf("%d%d%d",&x,&y,&v);
			map[x][y]=map[y][x]=v;
		}
		Dijkstra(0);
		
		for(int i=0;i<n;i++)
			cout<<dis[i]<<" ";
		cout<<endl;
		memset(dep,0,sizeof(dep));
		int res=DFS(2);
		cout<<"2到0的最短路径条数为:"<<res<<endl;
	}
	return 0;
}