洛谷—— P2196 挖地雷

https://www.luogu.org/problem/show?pid=2196

题目背景

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


建立超级源点s，从s跑一边最短路，记录每个在最短路上的点的前驱、
处理一下输出便可、

#include <cstdio>
#include <queue>

#define max(a,b) (a>b?a:b)
inline void read(int &x)
{
    x=0; register char ch=getchar();
    for(; ch>'9'||ch<'0'; ) ch=getchar();
    for(; ch>='0'&&ch<='9'; ch=getchar()) x=x*10+ch-'0';
}
const int INF(0x3f3f3f3f);
const int N(26);
int val[N],head[N],sumedge;
struct Edge {
    int v,next,w;
    Edge(int v=0,int next=0,int w=0):v(v),next(next),w(w){}
}edge[N*N];
inline void ins(int u,int v,int w)
{
    edge[++sumedge]=Edge(v,head[u],w); head[u]=sumedge;
}

bool inq[N];
int dis[N],pre[N],ans[N];
std::queue<int>que;
inline void SPFA(int s)
{
    que.push(s);
    for(int u,v; !que.empty(); )
    {
        u=que.front(); que.pop(); inq[u]=0;
        for(int i=head[u]; i; i=edge[i].next)
        {
            v=edge[i].v;
            if(dis[v]<dis[u]+edge[i].w)
            {
                pre[v]=u;
                dis[v]=dis[u]+edge[i].w;
                if(!inq[v]) inq[v]=1,que.push(v);
            }
        }
    }
}

int Presist()
{
    int n; read(n);
    for(int i=1; i<=n; ++i)
        read(val[i]),ins(0,i,val[i]);
    for(int i=1,x; i<n; ++i)
      for(int j=i+1; j<=n; ++j)
        { read(x); if(x) ins(i,j,val[j]); }
    SPFA(0); int maxdis=0,t,cnt=0;
    for(int i=1; i<=n; ++i)
        if(maxdis<dis[i]) maxdis=dis[t=i];
    for(; t; t=pre[t]) ans[++cnt]=t;
    for(int i=cnt; i; --i) printf("%d ",ans[i]);
    printf("
%d",maxdis);
    return 0;
}

int Aptal=Presist();
int main(int argc,char*argv[]){;}