[深度优先搜索] 宝藏收集

描述

你是一名探险家，进入了一片神秘的森林。森林中有多个藏宝点，每个藏宝点中都埋有一定数量的宝藏。同时，藏宝点之间有许多路径连接，你可以沿着这些路径进行探险。给出藏宝点及其连接的数据后，你可以从任一处开始探险，然后沿着指定的路径继续前行（仅能选择一条路径），当无路径可走时，探险结束。设计一个探险方案，使你能收集到最多的宝藏。

在一个森林中有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;
}