题目链接
典型的树形dp。
设 \(dp(x, i)\) 表示 \(x\) 的子树内逗留 \(i\) 秒的作品最大值。
\[dp(x, i)=\max_{y\in x}\max_{i=0}^s\max_{j=2\times z}^i dp(y,j-2\times z)-dp(x,j-i) \]实际实现时有一些先后顺序和细节,调整一下即可。
建边方面可以用一个栈来实现。
code
// Problem: P1270 “访问”美术馆
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1270
// Memory Limit: 125 MB
// Time Limit: 1000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<bits/stdc++.h>
using namespace std;
#define int long long
inline int read(){int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;
ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+
(x<<3)+(ch^48);ch=getchar();}return x*f;}
#define M 600
//#define mo
#define N 510
struct node
{
int x, y, z, n;
}d[N*4];
int n, m, i, j, k;
int a[N], dp[N][M], h[N], f[M];
int zhan[N], top, x, y, s;
void cun(int x, int y, int z)
{
// printf("%lld %lld %lld\n", x, y, z);
++m;
d[m].x=x; d[m].y=y; d[m].z=z;
d[m].n=h[x]; h[x]=m;
}
void dfs(int x)
{
if(a[x])
{
for(int i=0; i<=a[x]&&i*5<=s; ++i)
dp[x][i*5]=i;
for(int i=1; i<=s; ++i)
if(!dp[x][i]) dp[x][i]=dp[x][i-1];
// printf("dp[%lld]=%lld\n", x, dp[x][s]);
return ;
}
for(int g=h[x]; g; g=d[g].n)
{
int y=d[g].y;
dfs(y);
// memset(f, 0, sizeof(f));
for(int j=0; j<=s; ++j) f[j]=dp[x][j];
for(int j=s; j>=0; --j)
for(int i=j; i>=d[g].z*2; --i)
dp[x][j]=max(dp[x][j], dp[y][i-d[g].z*2]+f[j-i]);
}
// printf("dp[%lld]=%lld\n", x, dp[x][s]);
}
signed main()
{
// freopen("tiaoshi.in", "r", stdin);
// freopen("tiaoshi.out", "w", stdout);
s=read()-1;
zhan[top=1]=k=1;
while(scanf("%lld%lld", &x, &y)!=EOF)
{
if(y==0) cun(zhan[top--], ++k, x), zhan[++top]=k, zhan[++top]=k;
else cun(zhan[top--], ++k, x), a[k]=y;
}
dfs(1);
printf("%lld", dp[1][s]);
return 0;
}