P2196 挖地雷

题目描述

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

输入格式

有若干行。

第11行只有一个数字，表示地窖的个数NN。

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

第33行至第N+1N+1行表示地窖之间的连接情况：

第33行有n-1n−1个数（00或11），表示第一个地窖至第22个、第33个、…、第nn个地窖有否路径连接。如第33行为1 1 0 0 0 … 011000…0，则表示第11个地窖至第22个地窖有路径，至第33个地窖有路径，至第44个地窖、第55个、…、第nn个地窖没有路径。

第44行有n-2n−2个数，表示第二个地窖至第33个、第44个、…、第nn个地窖有否路径连接。

… …

第n+1n+1行有11个数，表示第n-1n−1个地窖至第nn个地窖有否路径连接。（为00表示没有路径，为11表示有路径）。

输出格式

有两行

第一行表示挖得最多地雷时的挖地雷的顺序，各地窖序号间以一个空格分隔，不得有多余的空格。

第二行只有一个数，表示能挖到的最多地雷数。

输入输出样例

输入 #1复制

5
10 8 4 7 6
1 1 1 0
0 0 0
1 1
1

输出 #1复制

1 3 4 5
27

 

思路：这是dp入门题。

dp[i]:表示以第i个洞结尾做多有几个地雷。

dp[i]=max(dp[i],dp[j]+dp[i]);  其中（i,j）有连接。

最后去dp[1-n]中最大的一个。

此题需要记录路径，用一个数组记录前驱，递归输入即可。

#include <bits/stdc++.h>
using namespace std;


int f[50],a[50],mark[50][50],path[50],g[50];

void oprint(int x){//递归输出
    if(path[x]==-1){ //前驱为-1说明找到路径的开端了
       cout<<x<<" ";
       return;
    }
    oprint(path[x]);//先输出前驱
    cout<<x<<" ";
}
int main()
{
    int n,x,id;
    cin>>n;
    for(int i=1; i<=n; i++){
       cin>>a[i];
    }
    for(int i=1; i<n; i++){
       for(int j=i+1; j<=n; j++){
         cin>>x;
         mark[i][j]=mark[j][i]=x;
       }
    }
    memset(path,-1,sizeof(path));
    int ans=0;
    for(int i=1; i<=n; i++){//f[i]表示以i为结尾的地雷个数
        f[i]=a[i];//至少自己的洞有a[i]个
        int ma=0,k=0;
        for(int j=1; j<=n; j++){//找到一个最大的f[j]
           if(i!=j&& mark[i][j]&&f[j]>ma){
                  path[i]=j;//记录i的前驱
                  ma=f[j];
                  k=1;
           }
        }
        if(k) f[i]+=ma;
        if(ans<f[i]){
            ans=max(f[i],ans);
            id=i;
        }
    }
    oprint(id);
    cout<<endl;
    cout<<ans<<endl;
    return 0;
}
//拓扑排序