挖地雷

挖地雷

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

Input

第1行：1个整数N，表示地窖的个数
第2行：N个用空格分开的整数，第i个整数表示地窖i里所埋的地雷个数
第3行~N+1行：第i+2行有n-i个用空格分开的整数，第i+2行描述第i个地窖与第i+1,i+2,...,n个地窖的连接情况。某个整数为1，表示有路径相连，为0，表示无路径相连。

Output

第1行：1个整数，表示能挖到的最多的地雷数量
第2行：若干个用空格分开的整数，表示所挖的地窖编号。编号按所挖地窖的顺序输出，若有多个方案都得到最多的地雷数量，则输出编号尽可能小的一组。

Sample Intput

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

Sample Output

31
1 2 4 5

Range

1<=N<=20;

Analysis

因为挖地雷时只能选择一条路径，且不能倒着挖，所以满足“无后效性”，所以可以使用动态规划(也可以用Dfs或Bfs)的算法来解题。
所以定义Max数组来储存从当前地窖开始往后挖能挖到的最多的地雷数。
定义Next数组记录下一个地窖的位置。
定义Sum数组储存每个地窖的地雷数。
定义Roud二维数组储存路径。Roud[i][j]表示从i到j是否有路。

Code

#include<cstdio>
#include<algorithm>
using namespace std;
int Sum[25],Roud[25][25],Max[25],Next[25];
int l,k,maxx=-999999999;
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&Sum[i]);
    for(int i=1;i<n;i++)
        for(int j=i+1;j<=n;j++)
            scanf("%d",&Roud[i][j]);
    Max[n]=Sum[n];//因为是最后一个地窖，所以从它挖起能挖到的最多的地雷数是它本身
    for(int i=n-1;i>=1;i--)
    {
        l=0;//下一个地窖的最优挖雷数
        k=0;//下一个地窖的位置
        for(int j=i+1;j<=n;j++)
            if(Roud[i][j]==1&&Max[j]>l)
            {
                l=Max[j];
                k=j;
            }
        Max[i]=l+Sum[i];//从第i个起能挖到的最多的地雷数
        Next[i]=k;//记录最优路径
    }
    for(int i=1;i<=n;i++)
        if(Max[i]>maxx)
            {
                k=i;
                maxx=Max[i];
            }//计算最多挖出的雷的个数
    printf("%d\n",Max[k]);
    printf("%d",k);//输出挖雷的顺序（这里处理空格）
    k=Next[k];//将下一个地窖位置赋值
    while(k!=0)
    {
        printf(" %d",k);
        k=Next[k];//将下一个地窖位置赋值
    }
    printf("\n");
    return 0;
}