洛谷P1270 “访问”美术馆

题目描述

经过数月的精心准备，Peer Brelstet，一个出了名的盗画者，准备开始他的下一个行动。艺术馆的结构，每条走廊要么分叉为两条走廊，要么通向一个展览室。Peer知道每个展室里藏画的数量，并且他精确测量了通过每条走廊的时间。由于经验老到，他拿下一幅画需要5秒的时间。你的任务是编一个程序，计算在警察赶来之前，他最多能偷到多少幅画。

输入格式

第1行是警察赶到的时间，以s为单位。第2行描述了艺术馆的结构，是一串非负整数，成对地出现：每一对的第一个数是走过一条走廊的时间，第2个数是它末端的藏画数量；如果第2个数是0，那么说明这条走廊分叉为两条另外的走廊。数据按照深度优先的次序给出，请看样例。

一个展室最多有20幅画。通过每个走廊的时间不超过20s。艺术馆最多有100个展室。警察赶到的时间在10min以内。

输出格式

输出偷到的画的数量

输入输出样例

输入 #1

60
7 0 8 0 3 1 14 2 10 0 12 4 6 2

输出 #1

2

思路：还是比较明显的树上背包题目，主要就是在输入边的时候是按照深度优先搜索给出的，这配合一个stack来读取则很好；至于其他的注意点就是这里的size是在于每个子树的来回时间以及偷东西需要的时间（也是离散的，记得处理好），由于边来回时间，不再是经典的dp[u][i+j]=max(dp[u][i+j],dp[u][i]+dp[y][j])了，而是dp[u][i+j+2*w]=max(dp[u][i+j+2*w],dp[u][i]+dp[y][j])，由于从u到v的来回还要2*w的时间；还有就是输出的是要警察到达时间-1的dp值，否则还是会被抓！

#include<iostream>
#include<algorithm>
#include<map>
#include<cmath>
#include<cstring>
#include<iomanip>
#include<numeric>
#include<stack>
#include<queue>
#include<set>
#include<string>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f,maxn=1e4+5;//结点数量
//树上背包
//二叉树结构，就是到树根还有个时间
//题目说是按照深度优先遍历的顺序给出，搞个stack比较好办事
int arr[maxn]={},vcnt[maxn]={1},dp[maxn][605]={},cnt=0,lim;//存量，访问次数等
//dp[u][i]表示以u结点为根的子树能在i秒内偷到多少，且i算来回和偷的时间的
struct edge{
	int v,w;
};
vector<edge> g[maxn];
stack<int> stk;
int dfs(int u)
{
	if(arr[u])
	{
		for(int i=1;i<=arr[u];i++)
		{
			dp[u][5*i]=i;
		}
		return arr[u]*5;
	}
	int p=0;//抽象了，是总的最多能游走的时间
	for(auto e:g[u])
	{
		int v=e.v,w=e.w;
		int sz=dfs(v);
		for(int i=min(lim,p);i>=0;i--)
			for(int j=0;j<=sz&&i+j+2*w<=lim;j++)
				dp[u][i+j+2*w]=max(dp[u][i+j+2*w],dp[u][i]+dp[v][j]);
		p+=2*w+sz;
	}
	return p;
}
int main()
{
	ios_base::sync_with_stdio(0),cin.tie(0);
	cin>>lim;//警察时间
	stk.push(cnt++);
	while(!stk.empty())
	{
		int u=stk.top();
		if(vcnt[u])stk.pop();
		else vcnt[u]++;//这样就实现每个点搞两遍了
		int v=cnt++,w,num;
		cin>>w>>num;
		g[u].push_back({v,w});
		if(!num)stk.push(v);
		else arr[v]=num;
	}
	dfs(0);
	cout<<dp[0][lim-1];
	return 0;
}