luogu2196 挖地雷 dp

http://www.elijahqi.win/archives/466
题目背景
NOIp1996提高组第三题
题目描述
在一个地图上有N个地窖（N<=20），每个地窖中埋有一定数量的地雷。同时，给出地窖之间的连接路径。当地窖及其连接的数据给出之后，某人可以从任一处开始挖地雷，然后可以沿着指出的连接往下挖（仅能选择一条路径），当无连接时挖地雷工作结束。设计一个挖地雷的方案，使某人能挖到最多的地雷。
输入输出格式
输入格式：

输入文件mine.in有若干行。
第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表示有路径）。

输出格式：

输出文件wdl.out有两行数据。
第一行表示挖得最多地雷时的挖地雷的顺序，各地窖序号间以一个空格分隔，不得有多余的空格。
第二行只有一个数，表示能挖到的最多地雷数。

输入输出样例
输入样例#1：
5
10 8 4 7 6
1 1 1 0
0 0 0
1 1
1
输出样例#1：
1 3 4 5
27

因为是单向的，所以从第n-1个地雷坑开始判断，用自己的地雷数加上本坑能通到的所有坑中地雷数最多的。这样一直倒推回去，就能AC了。

#include<cstdio>
#define N 25
int n,w[N],map[N][N],f[N],st[N];
int main(){
    freopen("2196.in","r",stdin);
    scanf("%d",&n);
    for (int i=1;i<=n;++i) scanf("%d",&w[i]),f[i]=w[i];
    for (int i=1;i<n;++i)
        for (int j=i+1;j<=n;++j) scanf("%d",&map[i][j]);
    int tmp=0,ans=0;
    for (int i=n-1;i>=1;--i){
        for (int j=i+1;j<=n;++j)
            if (map[i][j]&&w[i]+f[j]>f[i]) f[i]=w[i]+f[j],st[i]=j;
    }
    for (int i=1;i<=n;++i) if (f[i]>f[ans]) ans=i;
    tmp=ans; 
    while (st[tmp]) printf("%d ",tmp),tmp=st[tmp];
    printf("%d\n%d",tmp,f[ans]);
    return 0;
}