[DP 路径记录]P2196 [NOIP1996 提高组] 挖地雷

[DP 路径记录]P2196 [NOIP1996 提高组] 挖地雷

题目

思路

挺水的题目，主要是利用前驱记录转移路径

代码

// Problem: P2196 [NOIP1996 提高组] 挖地雷
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P2196
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// FishingRod

#include<bits/stdc++.h>
using namespace std;
#define endl "\n"
typedef long long LL;
typedef pair<int,int> PII;
//#define MULINPUT
/*DATA & KEY

*/
int T;
const int N=25;
LL f[N],w[N];
int pre[N],path[N];
bool g[N][N];
void solve(int C)
{
	//NEW DATA CLEAN
	
	//NOTE!!!
	memset(pre,-1,sizeof pre);
	int n;cin>>n;
	for(int i=1;i<=n;i++)cin>>w[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++)f[i]=w[i];
	for(int i=1;i<=n;i++)
		for(int j=1;j<i;j++)
			if(i!=j&&g[j][i])
				if(f[i]<f[j]+w[i])
				{
					f[i]=f[j]+w[i];
					pre[i]=j;
				}
	LL ans=0,ed=0;
	for(int i=1;i<=n;i++)
		if(f[i]>ans)
		{ans=f[i];ed=i;}
	int cnt=0;
	for(int i=ed;i!=-1;i=pre[i])path[++cnt]=i;
	for(int i=cnt;i>=1;i--)cout<<path[i]<<" ";
	cout<<endl<<ans<<endl;
}

int main()
{
	#ifdef MULINPUT
		scanf("%d",&T);
		for(int i=1;i<=T;i++)solve(i);
	#else
		solve(1);
	#endif
	return 0;
}