洛谷2607 骑士（基环树+树形DP）

传送门

【题目分析】

第一眼：咦这不简单树形DP吗？

第二眼：嗯？这不是有N条边吗？怎么就树形DP了？

第三眼：唉好像拆一条边不就N-1条边了吗？哎嘿嘿我太聪明了。。。。

噼里啪啦打完一交，WA完。。。。。。。一脸懵？？？才发现可能直接将整个图（以为保证连通）拆成两个联通块了。。。。。

然后画画图，发现肯定是拆环上的边，搞个并查集记录一下好了。

然后对于拆掉的边，直接分别强制两端点不选做一遍树形DP即可，取个较大值。考虑DP，每个点就两个选择：选和不选。如果不选，那么当前点最大值就为所有儿子的最大值之和；如果选，那么当前点最大值就为所有儿子的不选的最大值之和。

啪啪啪改完又交上去，对了三个点，一脸懵。

可能有多个联通块！多个！！！情况不考虑完真的难受

所以就可能要多次拆边，要开数组记录。每组都要做一遍，最后答案为各组最大值之和。

【代码~】

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int MAXN=1e6+10;
const int MAXM=2e6+10;

int n,cnt;
int s[MAXN],t[MAXN],tot;
LL ans;
int head[MAXN];
int nxt[MAXM],to[MAXM];
int a[MAXN];
LL dpfa[MAXN],dpson[MAXN];
int fa[MAXN];

int Read(){
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

int find(int x){
	if(x==fa[x])
	  return x;
	return fa[x]=find(fa[x]);
}

void add(int x,int y){
	nxt[cnt]=head[x];
	head[x]=cnt;
	to[cnt]=y;
	cnt++;
}

void dfs(int u,int f){
	dpfa[u]=a[u],dpson[u]=0;
	for(int i=head[u];i!=-1;i=nxt[i]){
		int v=to[i];
		if(v==f)
		  continue;
		dfs(v,u);
		dpson[u]+=max(dpson[v],dpfa[v]);
		dpfa[u]+=dpson[v];
	}
}

int main(){
	memset(head,-1,sizeof(head));
	n=Read();
	for(int i=1;i<=n;++i){
		fa[i]=i;
	}
	for(int i=1;i<=n;++i){
		a[i]=Read();
		int y=Read();
		if(find(i)==find(y)){
			s[++tot]=i;
			t[tot]=y;
		}
		else{
			add(i,y),add(y,i);
			fa[fa[i]]=fa[y];
		}
	}
	LL tt;
	for(int i=1;i<=tot;++i){
		dfs(s[i],-1);
		tt=dpson[s[i]];
		dfs(t[i],-1);
		ans+=max(tt,dpson[t[i]]);
	}
	cout<<ans;
	return 0;
}