PTA直捣黄龙(STL中容器的灵活运用和DFS)
题目:
本题是一部战争大片 —— 你需要从己方大本营出发,一路攻城略地杀到敌方大本营。首先时间就是生命,所以你必须选择合适的路径,以最快的速度占领敌方大本营。当这样的路径不唯一时,要求选择可以沿途解放最多城镇的路径。若这样的路径也不唯一,则选择可以有效杀伤最多敌军的路径。
输入格式:
输入第一行给出 2 个正整数 N(2 ≤ N ≤ 200,城镇总数)和 K(城镇间道路条数),以及己方大本营和敌方大本营的代号。随后 N-1 行,每行给出除了己方大本营外的一个城镇的代号和驻守的敌军数量,其间以空格分隔。再后面有 K 行,每行按格式城镇1 城镇2 距离
给出两个城镇之间道路的长度。这里设每个城镇(包括双方大本营)的代号是由 3 个大写英文字母组成的字符串。
输出格式:
按照题目要求找到最合适的进攻路径(题目保证速度最快、解放最多、杀伤最强的路径是唯一的),并在第一行按照格式己方大本营->城镇1->...->敌方大本营
输出。第二行顺序输出最快进攻路径的条数、最短进攻距离、歼敌总数,其间以 1 个空格分隔,行首尾不得有多余空格。
输入样例:
10 12 PAT DBYDBY 100PTA 20PDS 90PMS 40TAP 50ATP 200LNN 80LAO 30LON 70PAT PTA 10PAT PMS 10PAT ATP 20PAT LNN 10LNN LAO 10LAO LON 10LON DBY 10PMS TAP 10TAP DBY 10DBY PDS 10PDS PTA 10DBY ATP 10
输出样例:
PAT->PTA->PDS->DBY3 30 210
分析:
找合适的数据结构是个难点,由于图中的点是字符串,所以也不能用一般形式下的二维数组或者领接表存图了,本题测试点较弱,故dfs也可以幸运的AC。
#includeusing namespace std;mapcityman;//存驻守的敌军数量 map<string,vector<pair>>mp;//类似于图的领接表存图 string s,e;//s,e分别表示起点和终点mapvis;//用于判断该城市有没有来过 //INT_MAX和INT_MIN是在头文件#include下的,INT_MAX = 2^31-1=2147483647,INT_MIN= -2^31=-2147483648 int minlen=INT_MAX,maxnum=INT_MIN,maxman=INT_MIN;//minlen最短进攻距离,maxnum解放最多城市数,maxnum最多歼敌数vectorpath,bpath;//path为临时的路径,bpath为最终的路径 int cnt;//最快进攻路径的条数void dfs(string s,int len,int num,int man){//s:当前要访问的顶点,len,num,man:从起点到s这条当前路径的长度,城市数和人数if(s==e)//到终点,找到一条新路径{if(lenmaxnum||(num==maxnum&&man>maxman)){//城市更多或者城市相同但消灭人数更多maxnum=num,maxman=man;bpath=path;}}return;}for(auto x:mp[s]){if(!vis[x.first])//本路径尚未经过该顶点{vis[x.first]=1;path.push_back(x.first);dfs(x.first,len+x.second,num+1,man+cityman[x.first]);vis[x.first]=0;//恢复vis和pathpath.pop_back();}}}int main(){int n,m;cin>>n>>m>>s>>e;while(--n){string t;int x;cin>>t>>x;cityman[t]=x;}while(m--){string t,u;int x;cin>>t>>u>>x;mp[t].push_back({u,x});mp[u].push_back({t,x});}path.push_back(s);dfs(s,0,0,0);int flag=0;for(auto x:bpath){if(flag) cout<";flag=1;cout<<x;}cout<<endl<<cnt<<" "<<minlen<<" "<<maxman; return 0;}