【一只蒟蒻的刷题历程】 【洛谷】 P2196-挖地雷 （记忆化搜索）

题目描述

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

输入格式

有若干行。

第11行只有一个数字，表示地窖的个数NN。

第22行有NN个数，分别表示每个地窖中的地雷个数。

第33行至第N+1N+1行表示地窖之间的连接情况：

第33行有n-1n−1个数（00或11），表示第一个地窖至第22个、第33个、…、第nn个地窖有否路径连接。如第33行为1 1 0 0 0 … 011000…0，则表示第11个地窖至第22个地窖有路径，至第33个地窖有路径，至第44个地窖、第55个、…、第nn个地窖没有路径。

第44行有n-2n−2个数，表示第二个地窖至第33个、第44个、…、第nn个地窖有否路径连接。

… …

第n+1n+1行有11个数，表示第n-1n−1个地窖至第nn个地窖有否路径连接。（为00表示没有路径，为11表示有路径）。

输出格式

有两行

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

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

输入输出样例

输入 #1

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

输出 #1复制

1 3 4 5
27

---

思路：

因为不知道从第几个洞出发最快，索性for一遍，找到最大的dp**（地雷数）以及每个点往后走哪更优（路径）**。因为数据不大，而且是记忆化，所以用的时间不多（后面的点可能在前面的路径中就更新了）。

思想：dp【i】 = max(dp【i】，dp【j】+boom【i】)

主要就是：从某点出发的各条路径的最大值
（每一层都是这个思想，直到往后没有路径，开始返回结果并更新这条路径中每个位置的最优值）

其中，j 指的是 i 可以走的下一个点，boom为当前洞的地雷数

---

代码：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int n,x,boom[25],ans=0,p;//boom地雷数
vector<int> g[25];  //邻接表
int dp[25]={0},path[25]; //dp数组和路径数组
int r(int now)
{
	if(dp[now]) return dp[now]; //计算过，直接返回
	int v=0; //也就是找到上面 **思想** 中的 j
    for(int i=0;i<g[now].size();i++)
    {
    	int u=g[now][i];
    	int value=r(u)+boom[now]; //**思想** 中 max的后半部分
    	if(value>dp[now])
    	{
    	    dp[now]=value; //更新
    	    v=u;
		}
	}
	path[now]=v; //路径
	return dp[now]; //返回结果
}
int main() 
{
	fill(path,path+25,-1);
    cin>>n;
	for(int i=1;i<=n;i++)  cin>>boom[i]; //地雷数

	for(int i=1;i<n;i++)
	{
		for(int j=i+1;j<=n;j++) 
	  {
		 cin>>x; 
	   	 if(x) g[i].push_back(j); //邻接表
	  }
	  if(!g[i].size()) dp[i]=boom[i]; //后面没路径，最优值就是自己地雷数
	}
    dp[n]=boom[n]; //最后一个点，后面也没有路径，也是自己的地雷数
    
    for(int i=1;i<=n;i++) if(dp[i]==0) r(i); //跑一遍搜索

    for(int i=1;i<=n;i++)
    	if(dp[i]>ans) ans=dp[i],p=i; //ans记录最大值，p记录路径的开头
    cout<<p;
	while(path[p]!=-1)  //不断往后，直到-1
	{
		cout<<" "<<path[p];
		p=path[p];
	}
	cout<<endl<<ans; 
	return 0;
}