【洛谷】P2196 挖地雷 + dfs

【洛谷】P2196 挖地雷 + dfs

题目描述
在一个地图上有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 输出 #1
5 1 3 4 5
10 8 4 7 6 27
1 1 1 0
0 0 0
1 1
1

虽然我也不知道为什么这个题会放在动态规划里，但是我第一感觉就是使用深度优先搜索，这个和Dijkstra算法又有相似，想了解一下可以看一下这篇文章，有好处的。Dijkstra（单源最短路径）
这个题无非就是没有注明从那开始，那么就只能把每个点都进行dfs然后刷新路径和最值。dfs思路就不说了，代码上可以看的明白，祝各位早日AC
代码有点不规范哈

#include<bits/stdc++.h>
using namespace std;
int n,cnt,maxx,g[100][100],a[100],book[100],path[100],ans[100];
//   路径点数  到达数组   地雷数组 标记数组 路径数组  最大地雷时的路径数组 
int check(int k)
{
	for(int i=1;i<=n;i++)
	{
		if(!book[i]&&g[k][i])
		return 0;
	}
	return 1;
}
void dfs(int k,int stp,int sum)
{
	if(check(k))
	{
		if(sum>maxx)
		{
			cnt=stp;  //刷新 
			maxx=sum; //刷新 
			for(int i=1;i<=stp;i++)ans[i]=path[i];  //路径数组赋值 
		}
		return;
	}
	for(int i=1;i<=n;i++)
	{
		if(g[k][i]&&!book[i])
		{
			book[i]=1;
			path[stp+1]=i;
			dfs(i,stp+1,sum+a[i]);
			book[i]=0;    //回溯 
		}
	}
	
}

int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<n;i++)
	for(int j=i+1;j<=n;j++)
	cin>>g[i][j];
	for(int i=1;i<n;i++)
	{
		book[i]=1;
		path[1]=i;
		dfs(i,1,a[i]);
		book[i]=0;    //回溯 
	}
	for(int i=1;i<=cnt;i++)
    cout<<ans[i]<<" ";
    cout<<endl<<maxx;
    return 0;
}