题目链接:
思路:
典型的dp题,由于题目规定了只能由编号小的地窖到编号大的,所以dp顺序是确定的,不需要进行拓扑排序,只需要判断 i,j 之间是否有连接即可。用 f[ i ]表示以i结尾的挖地雷路径的最大挖地雷数,用pre来记录路径前驱,dfs打印路径。
代码:
#include <bits/stdc++.h>
using namespace std;
const int maxn = 21;
int n, arr[maxn];
int g[maxn][maxn];
int f[maxn]; //以i为终点能挖到的最多地雷数
int pre[maxn];
void print(int i){
if(pre[i]==i) {cout<<i<<' '; return;}
print(pre[i]);
cout << i << ' ';
}
int main(){
cin >> n;
for(int i=1; i<=n; i++){
cin>>arr[i];
f[i]=arr[i];
pre[i] = i;
}
for(int i=1; i<=n-1; i++)
for(int j=i+1; j<=n; j++)
cin >> g[i][j];
for(int i=2; i<=n; i++){
for(int j=1; j<i; j++){
if(!g[j][i]) continue;
if(f[i] < f[j]+arr[i]){
f[i] = f[j]+arr[i];
pre[i] = j;
}
}
}
int maxVal = -0x3f3f3f3f, maxIdx;
for(int i=1; i<=n; i++){
if(f[i] > maxVal){
maxVal = f[i];
maxIdx = i;
}
}
print(maxIdx); cout << '\n';
cout << maxVal;
}