[HNOI/AHOI2018]道路
Description:
W 国的交通呈一棵树的形状。W 国一共有\(n - 1\)个城市和\(n\)个乡村,其中城市从\(1\)到\(n - 1\) 编号,乡村从\(1\)到\(n\)编号,且\(1\)号城市是首都。道路都是单向的,本题中我们只考虑从乡村通往首都的道路网络。对于每一个城市,恰有一条公路和一条铁路通向这座城市。对于城市i, 通向该城市的道路(公路或铁路)的起点,要么是一个乡村,要么是一个编号比\(i\)大的城市。 没有道路通向任何乡村。除了首都以外,从任何城市或乡村出发只有一条道路;首都没有往 外的道路。从任何乡村出发,沿着唯一往外的道路走,总可以到达首都。
W 国的国王小 W 获得了一笔资金,他决定用这笔资金来改善交通。由于资金有限,小 W 只能翻修\(n - 1\)条道路。小 W 决定对每个城市翻修恰好一条通向它的道路,即从公路和铁 路中选择一条并进行翻修。小 W 希望从乡村通向城市可以尽可能地便利,于是根据人口调 查的数据,小 W 对每个乡村制定了三个参数,编号为\(i\)的乡村的三个参数是\(a_i\),\(b_i\)和\(c_i\)。假设 从编号为\(i\)的乡村走到首都一共需要经过\(x\)条未翻修的公路与\(y\)条未翻修的铁路,那么该乡村 的不便利值为
在给定的翻修方案下,每个乡村的不便利值相加的和为该翻修方案的不便利值。 翻修\(n - 1\)条道路有很多方案,其中不便利值最小的方案称为最优翻修方案,小 W 自然 希望找到最优翻修方案,请你帮助他求出这个最优翻修方案的不便利值。
Hint:
对于所有的数据,\(n \le 20000\),\(1 \le a_i,b_i \le 60\),\(1 \le c_i \le 10^9\),\(s_i,t_i\)是\([-n,-1] \cup (i,n - 1]\)内的整数,任意乡村可以通过不超过40条道路到达首都。
Solution:
表面上不可做实际上并不难的树型dp
设 \(f[i][j][k]\) 表示在i节点,到根修了j条公路,k条铁路的代价
显然有转移:
\(f[i][j][k]=min(f[ls][j+1][k]+f[rs][j][k],f[ls][j][k],f[rs][j][k+1])\)
可以考虑直接出叶子节点的f值,然后由根的初始状态出发\(f[1][0][0]\),记忆化搜索
但是这题卡空间,但是树的最大深度为40,我们发现一个点只由儿子转移过来,所以只需开一条链的空间大小,详见代码:
#include <map>
#include <set>
#include <stack>
#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define ls p<<1
#define rs p<<1|1
using namespace std;
typedef long long ll;
const int mxn=4e4+5;
int n,m,a[mxn],b[mxn],c[mxn],to[mxn][2];
ll f[105][85][85];
inline int read() {
char c=getchar(); int x=0,f=1;
while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}
while(c<='9'&&c>='0') {x=(x<<3)+(x<<1)+(c&15);c=getchar();}
return x*f;
}
inline int chkmax(int &x,int y) {if(x<y) x=y;}
inline int chkmin(int &x,int y) {if(x>y) x=y;}
struct ed {
int to,nxt;
}t[mxn<<1];
void dfs(int u,int id,int x,int y)
{
if(to[u][0]) dfs(to[u][0],id+1,x+1,y);
if(to[u][1]) dfs(to[u][1],id+2,x,y+1);
if(u>n) {
for(int i=0;i<=x;++i)
for(int j=0;j<=y;++j)
f[id][i][j]=1ll*c[u]*(a[u]+i)*(b[u]+j);
}
else {
for(int i=0;i<=x;++i)
for(int j=0;j<=y;++j)
f[id][i][j]=min(f[id+1][i+1][j]+f[id+2][i][j],f[id+1][i][j]+f[id+2][i][j+1]);
}
}
int main()
{
int u,v;
n=read(); memset(f,0x3f,sizeof(f));
for(int i=1;i<n;++i) {
u=read(); v=read();
if(u<0) u=-u+n;
if(v<0) v=-v+n;
to[i][0]=u; to[i][1]=v;
}
for(int i=1;i<=n;++i)
a[i+n]=read(),b[i+n]=read(),c[i+n]=read();
dfs(1,1,0,0);
printf("%lld",f[1][0][0]);
return 0;
}