欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

1003 Emergency

程序员文章站 2022-06-11 11:17:30
...

1003 Emergency1003 Emergency

题目大意:

有n个城市,每个城市都有一些救援队,你要从c1城市出发到c2救援,给你每座城市救援队的数量并给你m条路径及每条路径的长度。要求输出最短路径的数量以及最短路径中能带最多帮手的数量(路过那座城市就可以带走该城市的救援队)

解题思路:

一个简单的DFS图遍历就可以解决,因为图并不大。
代码如下:

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int n,m,c1,c2,cnt,mi=inf,sum;
int team[600];//每座城市的救援队
int ma[600][600];//图
int book[600];//标记那些城市已经被访问过,防止走回头路
void dfs(int cur,int len,int num)
{
  if(cur==c2)
  {
    if(len<mi)//更新最短路径以及救援队数量
    {
      mi=len;
      sum=1;
      cnt=num;
    }
    else if(len==mi)
    {
      sum++;
      if(num>cnt)cnt=num;
    }
    return;
  }
  for(int i=0;i<n;i++)
  {
    if(i!=cur&&ma[cur][i]!=inf&&book[i]==0)
    {
      book[i]=1;
      dfs(i,len+ma[cur][i],num+team[i]);
      book[i]=0;
    }
  }
  return ;
}
int main()
{
  std::ios::sync_with_stdio(false);
  cin.tie(0);
  //freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
  cin>>n>>m>>c1>>c2;
  meminf(ma);
  for(int i=0;i<n;i++) cin>>team[i];
  for(int i=1;i<=m;i++)
  {
    int a,b,c;
    cin>>a>>b>>c;
    ma[a][b]=ma[b][a]=c;
  }
  book[c1]=1;//标记起始城市
  dfs(c1,0,team[c1]);
  cout<<sum<<' '<<cnt<<endl;
  return 0;
}
相关标签: PAT甲级 DFS