PAT甲级练习题1003

采用深度优先搜索。

#include<iostream>
#include<vector>
#include<climits>
using namespace std;
//采用深度优先搜索

int minlen=INT_MAX; //最短路径
int num; //路径数
int maxteam; //最大的救援队数量
int N; //节点总数
int Len[500][500]={0};//城市之间距离
int team[500]={0}; //每个城市对应的救援队数量
bool S[500]; //表明当前城市是否被选用
void dfs(int node,const int end,int len,int mteam)
{
if(node==end) //当前节点在终点时
{
if(len<minlen)
{
minlen=len; //最短路径重新计数
maxteam=mteam; //总救援队重新计数
num=1; //最短路径个数重新计数
}
else
{
if(len==minlen) //多了一条路径选择
{
++num;
if(mteam>maxteam)
maxteam=mteam;
}
}
return;
}

if(len>minlen) //当前距离已大于最短距离时，放弃
{
return;
}
int p;
for(p=0;p<N;++p)
{
if(S[p] && Len[node][p]!=0) //当前节点未被选择且有通路时
{
S[p]=false;
dfs(p,end,len+Len[node][p],mteam+team[p]);
S[p]=true;
}
}
}

int main()
{
int M,C1,C2;
cin>>N>>M>>C1>>C2;

int i,j,k,L;

for(i=0;i<N;++i)
{
cin>>team[i];
S[i]=true;
}

for(k=0;k<M;++k)
{
cin>>i>>j>>L;
Len[i][j]=L;
Len[j][i]=L;
}
S[C1]=false;

dfs(C1,C2,0,team[C1]);

cout<<num<<" "<<maxteam<<endl;

return 0;
}