CCPC-Wannafly Winter Camp Day1 (Div2, onsite) E 流流流動(樹形dp)

流流流動

題解：題目是點的選與不選的問題，並且有連邊，因此我們很容易想到樹形$dp$，但是題目圖的並不是聯通的，因此我們可以將$0$與每一個連通集建邊。然後考慮$dp[u][1]$表示選取以點$u$爲根節點所能獲得的最大收益，$dp[u][0]$表示不選$u$作爲根節點所能獲取的最大收益。所以有$\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}$判斷是否有連邊用並查集即可。

代碼

#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;
}