虽然题面很长,但是静下心来看完后,还是掩盖不了这是一道水体的事实。
发现这是一棵完全二叉树,而且深度只有40。
那么dp[u][l][r]表示到达节点u,有 l 条公路,r条铁路没修。转移方程就是分两种情况递归下去就行啦。
然后中间节点只是用来记录答案的,在叶节点计算。
据说这道题卡空间,但是我并没有感觉。倒是因为计算的时候没int强制转换成long longWA了一发。
1 #include<cstdio> 2 #include<iostream> 3 #include<cmath> 4 #include<algorithm> 5 #include<cstring> 6 #include<cstdlib> 7 #include<cctype> 8 #include<vector> 9 #include<stack> 10 #include<queue> 11 using namespace std; 12 #define enter puts("") 13 #define space putchar(' ') 14 #define Mem(a, x) memset(a, x, sizeof(a)) 15 #define rg register 16 typedef long long ll; 17 typedef double db; 18 const int INF = 0x3f3f3f3f; 19 const db eps = 1e-8; 20 const int maxn = 2e4 + 5; 21 inline ll read() 22 { 23 ll ans = 0; 24 char ch = getchar(), last = ' '; 25 while(!isdigit(ch)) last = ch, ch = getchar(); 26 while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar(); 27 if(last == '-') ans = -ans; 28 return ans; 29 } 30 inline void write(ll x) 31 { 32 if(x < 0) x = -x, putchar('-'); 33 if(x >= 10) write(x / 10); 34 putchar(x % 10 + '0'); 35 } 36 37 int n; 38 int ls[maxn << 1], rs[maxn << 1]; 39 int a[maxn], b[maxn], c[maxn]; 40 41 ll dp[maxn][45][45]; 42 ll dfs(int now, int l, int r) 43 { 44 if(now < 0) return (ll)c[-now] * ((ll)a[-now] + l) * ((ll)b[-now] + r); 45 if(dp[now][l][r]) return dp[now][l][r]; 46 ll ret1 = dfs(ls[now], l + 1, r) + dfs(rs[now], l, r); 47 ll ret2 = dfs(ls[now], l, r) + dfs(rs[now], l, r + 1); 48 dp[now][l][r] = min(ret1, ret2); 49 return dp[now][l][r]; 50 } 51 52 int main() 53 { 54 n = read(); 55 for(int i = 1; i < n; ++i) ls[i] = read(), rs[i] = read(); 56 for(int i = 1; i <= n; ++i) a[i] = read(), b[i] = read(), c[i] = read(); 57 write(dfs(1, 0, 0)), enter; 58 return 0; 59 }