CCPC-Wannafly Winter Camp Day1 (Div2, onsite) E 流流流动

题目描述

喜欢数学的wlswls最近被萎住了。

现在他一共有1...n1...n这么多数字，取数字ii会得到f[i]f[i]的收益。数字之间有些边，对于所有的i(i != 1)i(i!=1)，若ii为奇数，则ii与3i+13i+1之间有边，否则ii与i/2i/2之间有边。如果一条边的两个顶点xyxy都被取了，那么会失去d[min(x, y)]d[min(x,y)]的价值。请问wlswls怎么取，才能使得收益最大？

输入描述

第一行一个整数nn。

接下来一行nn个整数表示ff。

接下来一行nn个整数表示dd。

1 \leq n \leq 1001≤n≤100

1 \leq f[i], d[i] \leq 10001≤f[i],d[i]≤1000

输出描述

输出一个整数表示答案。

样例输入 1 

2
10 10 
1 2

样例输出 1

19

首先建好边，边建好就剩树形dp了

判断两点之间有没有边   并查集

#include <bits/stdc++.h>
using namespace std;
const int maxn=210;
typedef long long ll;
int n,f[maxn],d[maxn];
int pre[maxn];
vector<int> G[maxn];
int find(int x){return pre[x]==0?x:pre[x]=find(pre[x]);}
void join(int x,int y)
{
	int fx=find(x),fy=find(y);
	if(fx!=fy) pre[fx]=fy;
}
int dp[maxn][2];

void dfs(int u,int fa)
{
	dp[u][0]=0;
	dp[u][1]=f[u];
	for(auto v:G[u])
	{
		if(v!=fa)
		{
			dfs(v,u);
			dp[u][0]+=max(dp[v][0],dp[v][1]);
			dp[u][1]+=max(dp[v][0],dp[v][1]-d[min(u,v)]);
		}
	}
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&f[i]);
	for(int i=1;i<=n;i++) scanf("%d",&d[i]);
	for(int i=2;i<=n;i++)
	{
		if(i&1)  //奇数 添加边
		{
			if(3*i+1<=n)
			{
				G[3*i+1].push_back(i);
				G[i].push_back(3*i+1);
				join(3*i+1,i);
		    }
		}
		else    //偶数 i/2添加边
		{   
			G[i].push_back(i/2);
			G[i/2].push_back(i);
			join(i,i/2);
		}
	}
	for(int i=1;i<=n;i++)
	{
		if(!pre[i]){G[0].push_back(i);G[i].push_back(0);}
	}
	dfs(0,0);
	printf("%d\n",max(dp[0][0],dp[0][1]));
	return 0;
}