p2196 挖地雷

题目描述

在一个地图上有NN个地窖(N \le 20)(N≤20)，每个地窖中埋有一定数量的地雷。同时，给出地窖之间的连接路径。当地窖及其连接的数据给出之后，某人可以从任一处开始挖地雷，然后可以沿着指出的连接往下挖（仅能选择一条路径），当无连接时挖地雷工作结束。设计一个挖地雷的方案，使某人能挖到最多的地雷。

输入格式

有若干行。

第1行只有一个数字，表示地窖的个数N。

第2行有N个数，分别表示每个地窖中的地雷个数。

第3行至第N+1N+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

代码：

#include <bits/stdc++.h>
#define eps 1e-15
using namespace std;
int dp[25][25],n,a[25][25],p[25],ans[25];//dp表示从i洞到j洞挖到的最大地雷数

void solve(){
    for(int i = 1; i <= n; i++){
        dp[i][i] = p[i];
        for(int j = i+1;j <= n; j++)
        if( a[j-1][j] )dp[i][j] = dp[i][j-1] + p[j];
        else {
            int sum = 0;
            for(int k = j-1; k>=i; k--)
            if( a[k][j] && dp[i][k] + p[j] > sum){sum = dp[i][k] + p[j];}
            //if(sum == 0)sum = p[i];如果这个洞和前面所有的都没连接那就置0
            dp[i][j] = sum;
        }
    }
}
void traceBack(int s, int e, int sum){
    int k=e;
    for(int i = e; i>=s; i--){
        if(sum == dp[s][i-1] + p[k] && a[i-1][k]) {ans[k] = 1;sum -= p[k];k=i-1;}
    }
    ans[s] = 1;
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)cin>>p[i];
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
        cin>>a[i][j];
    for(int i=1;i<=n;i++)a[i][i]=1;
    solve();
    int sum=0,s,e;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        if(dp[i][j]>sum){sum=dp[i][j];s=i;e=j;}
        traceBack(s,e,sum);
        for(int i=s;i<=e;i++)
            if(ans[i]==1)cout<<i<<" ";
        cout<<endl;
        cout<<sum;
    return 0;
}