描述
你是一名探险家,进入了一片神秘的森林。森林中有多个藏宝点,每个藏宝点中都埋有一定数量的宝藏。同时,藏宝点之间有许多路径连接,你可以沿着这些路径进行探险。给出藏宝点及其连接的数据后,你可以从任一处开始探险,然后沿着指定的路径继续前行(仅能选择一条路径),当无路径可走时,探险结束。设计一个探险方案,使你能收集到最多的宝藏。
在一个森林中有N(1 ≤ N ≤ 20)个藏宝点,每个藏宝点中埋有一定数量的宝藏。同时,给出藏宝点之间的连接路径。给出藏宝点及其连接的数据后,你可以从任一处开始探险,然后沿着指定的路径继续前行(仅能选择一条路径),当无路径可走时,探险结束。设计一个探险方案,使你能收集到最多的宝藏。
输入
输入由若干行组成。
1. 第1行只有一个数字,表示藏宝点的个数N。
2. 第2行有N个数,分别表示每个藏宝点中的宝藏数量。宝藏数量是正整数
3. 第3行至第N+1行表示藏宝点之间的连接情况:
(1) 第3行有 n-1 个数(0 或 1),表示第一个藏宝点至第 2 个、第 3 个 … 第 n 个藏宝点有无路径连接。如第 3 行为 1 1 0 0 0…0,则表示第 1 个藏宝点至第 2 个藏宝点有路径,至第 3 个藏宝点有路径,至第 4 个、第 5 个 … 第 n 个藏宝点没有路径。
(2) 第 4 行有 n-2 个数,表示第二个藏宝点至第 3 个、第 4 个 … 第 n 个藏宝点有无路径连接。
(3) ……
(4) 以此类推,第 N+1 行有 1 个数,表示第 n-1 个藏宝点至第 n 个藏宝点有无路径连接。(为 0 表示没有路径,为 1 表示有路径)。
注意:各个藏宝点之间为单向连接。编号高的点没有边连到编号低的点
输出
第一行表示探险顺序,各藏宝点序号间以一个空格分隔,不得有多余的空格。(最后一个序号后也有空格)
第二行只有一个数,表示能收集到的最多宝藏数。结果不超过109
如果有多条路径都可以获得最多宝藏数量,则输出字典序最小的那一条。例如,
序列 1 2 3 的字典序小于 1 10 12
样例输入
5 10 8 4 7 6 1 1 1 0 0 0 0 1 1 1
样例输出
1 3 4 5 27
解题分析
本题是很明显的一道深度优先搜索的题目,我们需要合理设计一个dfs递归函数。首先,我们要存储好每个地方的宝藏数量,以及整个图关系的连通顺序。这个时候我们还需要一些合理的观察,保证我们的输入。然后,由于本题需要记录最优路径,所以我们合理地使用vector这个容器,利用好它的push_back函数和erase函数,便于我们在递归的过程中实时更新。然后,我们发现需要比较路径之间的字典序,所以不妨全局重新定义vector中的小于“<”的比较运算。接着,我们在dfs中,首先给函数传入一个开始深度优先搜索的起点位置,然后我们设置一个flag作为标记,看看我们当前位置是否还能往下走,如果可以就把flag设置为1,如果不行的话就说明如果我们选择了这条路,那么当前的一个宝藏数量已经达到了最大了(注意在dfs的过程中我们同时也进行了回溯的操作)。我们就开始判断了,如果当前的价值更大,则我们记录下这条路径,如果价值和我们的最佳价值一致,那么我们就判断一下字典序,字典序小的作为我们的最终答案。最终输出我们选定的路径和得到的最佳宝藏数量即可。
代码实现
#include <iostream>
#include <cstdio>
#include <queue>
#include <bitset>
#include <map>
#include <set>
#include <algorithm>
#include <cstdlib>
#include <unordered_map>
#include <unordered_set>
#include <cmath>
#include <vector>
using namespace std;
int N;
int bz[21];
int reach[21][21]={0};
int ans=-1;
int nowValue=0;
vector<int> bestPath;
vector<int> nowPath;
bool operator<(const vector<int>& A, const vector<int>& B){
int lena=A.size();
int lenb=B.size();
for(int i=0;i<lena;i++)
for(int j=0;j<lenb;j++){
if(A[i]<B[i]) return true;
else if(A[i]>B[i]) return false;
}
return false;
}
void dfs(int k){
int flag=0;
for(int i=k+1;i<=N;i++){
if(reach[k][i]){
flag=1;
nowValue+=bz[i];
nowPath.push_back(i);
dfs(i);
nowValue-=bz[i];
nowPath.erase(nowPath.begin()+nowPath.size()-1);
}
}
if(flag==0){
if(nowValue>ans){
ans=nowValue;
bestPath.erase(bestPath.begin(),bestPath.end());
int len=nowPath.size();
for(int i=0;i<len;i++){
bestPath.push_back(nowPath[i]);
}
}
else if(nowValue==ans){
if(nowPath<bestPath){
bestPath.erase(bestPath.begin(),bestPath.end());
int len=nowPath.size();
for(int i=0;i<len;i++){
bestPath.push_back(nowPath[i]);
}
}
}
}
return;
}
int main(){
scanf("%d",&N);
for(int i=1;i<=N;i++){
scanf("%d",&bz[i]);
}
for(int i=1;i<=N-1;i++){
for(int j=i+1;j<=N;j++){
scanf("%d",&reach[i][j]);
}
}
for(int i=1;i<=N;i++){
nowValue=bz[i];
nowPath.push_back(i);
dfs(i);
nowPath.erase(nowPath.begin()+nowPath.size()-1);
}
int len=bestPath.size();
for(int i=0;i<len;i++){
if(i==0)
printf("%d",bestPath[i]);
else
printf(" %d",bestPath[i]);
}
printf("\n%d\n",ans);
return 0;
}