流流流动
题解:题目是点的选与不选的问题,并且有连边,因此我们很容易想到树形 d p dp dp,但是题目图的并不是联通的,因此我们可以将 0 0 0与每一个连通集建边。然后考虑 d p [ u ] [ 1 ] dp[u][1] dp[u][1]表示选取以点 u u u为根节点所能获得的最大收益, d p [ u ] [ 0 ] dp[u][0] dp[u][0]表示不选 u u u作为根节点所能获取的最大收益。所以有 { d p [ u ] [ 1 ] = d p [ u ] [ 1 ] + m a x ( d p [ v ] [ 1 ] − d [ m i n ( u , v ) ] , d p [ v ] [ 0 ] ) d p [ u ] [ 0 ] = d p [ u ] [ 0 ] + m a x ( d p [ v ] [ 1 ] , d p [ v ] [ 0 ] ) \begin{cases} dp[u][1] = dp[u][1] + max(dp[v][1] - d[min(u,v)],dp[v][0])\\\\ dp[u][0] = dp[u][0]+max(dp[v][1],dp[v][0]) \end{cases} ⎩⎪⎨⎪⎧dp[u][1]=dp[u][1]+max(dp[v][1]−d[min(u,v)],dp[v][0])dp[u][0]=dp[u][0]+max(dp[v][1],dp[v][0])判断是否有连边用并查集即可。
代码
#include<bits/stdc++.h>
#define P pair<int,int>
using namespace std;
int f[120],d[120],dp[120][2],father[120],n;
vector<int> e[120];
int getF(int v)
{
return father[v] == 0 ? v : father[v] = getF(father[v]);
}
void merge1(int u,int v)
{
int t1 = getF(u);
int t2 = getF(v);
if(t1 != t2) {
father[t2] = t1;
}
}
void dfs(int u,int pre)
{
dp[u][1] = f[u];
for(int i = 0; i < e[u].size(); ++i) {
int v = e[u][i];
if(v == pre) continue;
dfs(v,u);
dp[u][1] += max(dp[v][1] - d[min(u,v)],dp[v][0]);
dp[u][0] += max(dp[v][1],dp[v][0]);
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("input.in","r",stdin);
#endif
cin >> n;
for(int i = 1; i <= n; ++i) {
cin >> f[i];
}
for(int i = 1; i <= n; ++i) {
cin >> d[i];
}
for(int i = 2; i <= n; ++i) {
if(i & 1) {
if(3 * i + 1 <= n) {
e[i].push_back(3 * i + 1);
e[3 * i + 1].push_back(i);
merge1(i, 3 * i + 1);
}
}else{
e[i].push_back(i / 2);
e[i / 2].push_back(i);
merge1(i, i / 2);
}
}
for(int i = 1; i <= n; ++i) {
if(father[i] == 0) {
e[0].push_back(i);
e[i].push_back(0);
}
}
dfs(0,0);
cout << dp[0][0] << endl;
return 0;
}