洛谷—— 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：

输出样例#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=; register char ch=getchar();

     for(; ch>''||ch<''; ) ch=getchar();

     for(; ch>=''&&ch<=''; ch=getchar()) x=x*+ch-'';

 }

 const int INF(0x3f3f3f3f);

 const int N();

 int val[N],head[N],sumedge;

 struct Edge {

     int v,next,w;

     Edge(int v=,int next=,int w=):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]=;

         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]=,que.push(v);

             }

         }

     }

 }

 int Presist()

 {

     int n; read(n);

     for(int i=; i<=n; ++i)

         read(val[i]),ins(,i,val[i]);

     for(int i=,x; i<n; ++i)

       for(int j=i+; j<=n; ++j)

         { read(x); if(x) ins(i,j,val[j]); }

     SPFA(); int maxdis=,t,cnt=;

     for(int i=; 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("\n%d",maxdis);

     return ;

 }

 int Aptal=Presist();

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