2022.12.1洛谷P2196 [NOIP1996 提高组] 挖地雷

[NOIP1996 提高组] 挖地雷 原题链接:传送门

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

输入格式

有若干行。

第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表示有路径）。

输出格式

有两行

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

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

样例输入 

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

样例输出

1 3 4 5
27

可以看到上图其实本题的数学模型是个有向无环图。

这题有点像最长xx子序列，转化到这个语境内应该是最长连通子序列。

那么我们定义状态表示为，其意义为以i为终点能挖到的最大地雷数(包括i)

那么很显然状态转移方程如下：

那么怎么输出挖地雷的顺序呢？

我们可以定义数组来存储i的前驱节点，那么最后的输出就很容易了，递归一下即可。

有以下两种方式：

两种方式的核心思想都是递归

void dfs(int x){
	if(p[x]) dfs(p[x]);
	cout << x << " ";
	return;
}

while (k) {
	ans[++t] = k;
	k = q[k];
}

AC代码：

#include <iostream>
using namespace std;
int n, ans, k;
int a[25], g[25][25], f[25], p[25];

void dfs(int x){
	if(p[x]) dfs(p[x]);
	cout << x << " ";
	return;
}

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];
	}
	f[1] = a[1];
	for(int i = 2; i <= n; i++){
		f[i] = a[i];
		for(int j = i - 1; j >= 1; j--){
			if(g[j][i] && f[i] < f[j] + a[i]){
				f[i] = f[j] + a[i];
				p[i] = j;
			}
		}
		if(ans < f[i]){
			ans = f[i];
			k = i;
		}
	}
	dfs(k);
	cout << "\n" << ans << "\n";
	return 0;
}