题目描述:
在一个地图上有N个地窖(N≤20),每个地窖中埋有一定数量的地雷。同时,给出地窖之间的连接路径。当地窖及其连接的数据给出之后,某人可以从任一处开始挖地雷,然后可以沿着指出的连接往下挖(仅能选择一条路径),当无连接时挖地雷工作结束。设计一个挖地雷的方案,使某人能挖到最多的地雷。
输入格式:
有若干行。
第1行只有一个数字,表示地窖的个数N。
第2行有N个数,分别表示每个地窖中的地雷个数。
第3行至第N+1行表示地窖之间的连接情况:
第3行有n−1个数(0或1),表示第一个地窖至第2个、第3个、…、第n个地窖有否路径连接。如第3行为1 1 0 0 0 … 0,则表示第1个地窖至第2个地窖有路径,至第3个地窖有路径,至第4个地窖、第5个、…、第n个地窖没有路径。
第4行有n-2个数,表示第二个地窖至第3个、第4个、…、第n个地窖有否路径连接。
… …
第n+1行有1个数,表示第n−1个地窖至第n个地窖有否路径连接。(为0表示没有路径,为1表示有路径)。
输出格式:
有两行
第一行表示挖得最多地雷时的挖地雷的顺序,各地窖序号间以一个空格分隔,不得有多余的空格。
第二行只有一个数,表示能挖到的最多地雷数。
输入输出样例
输入:
5
10 8 4 7 6
1 1 1 0
0 0 0
1 1
1
输出:
1 3 4 5
27
AC代码(DFS):
#include<bits/stdc++.h>
using namespace std;
const int N = 21;
int data[N], vis[N], graph[N][N];
int n, ans = 0;
vector<int> v;
void dfs(int t, int sum, vector<int>& path) {
int tar = 0;
for (int i = 1; i <= n; i++) {
if (graph[t][i] == 1 && vis[i] == 0) tar++;
}
if (tar == 0) {
if (sum > ans) {
ans = sum;
v = path;
}
return ;
}
for (int i = 1; i <= n; i++) {
if (graph[t][i] == 1 && vis[i] == 0) {
vis[i] = 1;
path.push_back(i);
dfs(i, sum + data[i], path);
path.pop_back();
vis[i] = 0;
}
}
}
int main() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> data[i];
for (int i = 1; i <= n; i++) {
for (int j = i + 1; j <= n; j++)
cin >> graph[i][j];
}
for (int i = 1; i <= n; i++) {
memset(vis, 0, sizeof vis);
vector<int> a1;
a1.push_back(i);
dfs(i, data[i], a1);
}
for (int i = 0; i < v.size(); i++) {
if (i == 0) cout << v[i];
else cout << " " << v[i];
}
cout << endl;
cout << ans;
return 0;
}
AC代码(DP):
#include<bits/stdc++.h>
using namespace std;
const int N = 21;
int data[N], graph[N][N], f[N], pre[N];
int n, ans = 0, tar = 1;
//定义状态f[i]为以第i个节点结束的最大值
void showPath(int x) {
if (pre[x] == 0) {
cout << x;
return ;
}
showPath(pre[x]);
cout << " " << x;
}
int main() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> data[i];
for (int i = 1; i <= n; i++) {
for (int j = i + 1; j <= n; j++)
cin >> graph[i][j];
}
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= n; j++) {
if (graph[j][i] == 1 && f[j] > f[i]) {
f[i] = f[j];
pre[i] = j;
}
}
f[i] += data[i];
if (ans < f[i]) {
ans = f[i];
tar = i;
}
}
showPath(tar);
cout << endl;
cout << ans;
return 0;
}