挖地雷

题目描述

在一个地图上有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表示有路径）。

输出格式

有两行

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

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

输入输出样例

输入 #1

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

输出 #1

 

1 3 4 5
27

分析：
本题依然可以进行dfs，由于n<=20，dfs显然不会超时，故选取每一个点，对其下一个点进行dfs即可

CODE：

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int M=1005;
int n,a[M];
int head[M],next[M],to[M],tot;
int get(){
    int res=0,f=1;
    char c=getchar();
    while (c>'9'||c<'0') {
        if (c=='-') f=-1;
        c=getchar();
    }
    while (c<='9'&&c>='0'){
        res=(res<<3)+(res<<1)+c-'0';
        c=getchar();
    }
    return res*f;
}
void add(int u,int v){
    next[++tot]=head[u];
    head[u]=tot;
    to[tot]=v;
    return ;
}
int ans;
int f[M],ansl[M],cnt;
bool flag;
void dfs(int u,int tot,int now){
    if (tot>ans) {
        flag=true;
        ans=tot;
        for (int i=1;i<=now;i++) ansl[i]=f[i];
        cnt=now;
    }
    for (int i=head[u];i;i=next[i]){
        int v=to[i];
        f[now]=v;
        dfs(v,tot+a[v],now+1);
        f[now]=0;
    }
    return ;
}
int main(){
    n=get();
    for (int i=1;i<=n;i++) a[i]=get();
    for (int i=1;i<=n;i++){
        for (int j=i+1;j<=n;j++){
            int ok=get();
            if (ok) add(i,j);
        }
    }
    for (int i=1;i<=n;i++){
        flag=false;
        dfs(i,a[i],2);
        if (flag) ansl[1]=i,f[1]=i;
    }
    for (int i=1;i<cnt;i++) printf ("%d ",ansl[i]);
    printf ("
%d",ans);
    return 0;
}