5290: [Hnoi2018]道路

5290: [Hnoi2018]道路

链接

分析:

  注意题目中说每个城市翻新一条连向它的公路或者铁路,所以两种情况分别转移一下即可。

  注意压一下空间,最后的叶子节点不要要访问,空间少了一半。

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iostream>
#include<cctype>
#include<set>
#include<vector>
#include<queue>
#include<map>
#define fi(s) freopen(s,"r",stdin);
#define fo(s) freopen(s,"w",stdout);
using namespace std;
typedef long long LL; inline int read() {
int x=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
} const int N = ;
int head[N], ls[N], rs[N], a[N], b[N], c[N], siz0[N], siz1[N], n;
LL dp[][][]; inline LL Calc(int u,int x,int y) { // 从根到u的路径上,表示翻新了x条公路,y条铁路
if (u >= n) return 1ll * c[u] * (siz0[u] - x + a[u]) * (siz1[u] - y + b[u]);
return dp[u][x][y];
} void dfs(int u) {
if (u >= n) { return ;
// for (int i = 0; i <= siz0[u]; ++i)
// for (int j = 0; j <= siz1[u]; ++j)
// dp[u][i][j] = 1ll * c[u] * (siz0[u] - i + a[u]) * (siz1[u] - j + b[u]); // 表示翻新了i条公路,j条铁路
// return ;
}
siz0[ls[u]] = siz0[u] + ; siz1[ls[u]] = siz1[u]; dfs(ls[u]);
siz0[rs[u]] = siz0[u]; siz1[rs[u]] = siz1[u] + ; dfs(rs[u]); for (int i = ; i <= siz0[u]; ++i)
for (int j = ; j <= siz1[u]; ++j) {
dp[u][i][j] = 1e18;
dp[u][i][j] = min(dp[u][i][j], Calc(ls[u], i + , j) + Calc(rs[u], i, j)); // 翻新公路
dp[u][i][j] = min(dp[u][i][j], Calc(ls[u], i, j) + Calc(rs[u], i, j + )); // 翻新铁路
}
}
int main() {
n = read();
for (int i = ; i < n; ++i) {
int u = read(), v = read();
if (u < ) u = -u + n - ;
if (v < ) v = -v + n - ;
ls[i] = u;
rs[i] = v;
}
for (int i = ; i <= n; ++i)
a[i + n - ] = read(), b[i + n - ] = read(), c[i + n - ] = read();
dfs();
cout << dp[][][];
return ;
}
上一篇:C# 如何获取当前应用程序的父目录


下一篇:python相关资料链接