codeforces 1263F

题目链接

题意

给两棵有根树,叶子数相同,每个叶子都连接着某个电机,同一棵树的叶子连着不同的电机,电机数等于叶子数,问最多可以删掉几条边,使得每台电机至少连接着一个树根

数据范围

$叶子数num\le 1000,树的节点个数n,m\le 2000$

解法

dp

设val[0/1][l][r]表示第0/1棵树,连接[l,r]这段电机的边数之和

如果知道了这个,就可以$O(n^2)$递推出num个电机满足条件最多需要删几条边.
dp[j]=max(dp[j],dp[i-1]+max(val[0][i][j],val[1][i][j])),选择0或1号树,就是让另一棵树连接[l,r]这段的电机.

求val[0/1][l][r]的方法是dfs.

void dfs(int cnt,int u){
	if(u!=1)sz[u]=1;
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		dfs(cnt,v);sz[u]+=sz[v];
		l[u]=min(l[u],l[v]);
		r[u]=max(r[u],r[v]);
	}
	val[cnt][l[u]][r[u]]=max(val[cnt][l[u]][r[u]],sz[u]);
}
void solve(int cnt){
	n=read();
	for(int i=1;i<=n;i++)l[i]=n+1;
	for(int i=2;i<=n;i++)fa[i]=read();
	for(int i=1;i<=num;i++){
		pa[i]=read();
		l[pa[i]]=r[pa[i]]=i;
	}
	for(int i=2;i<=n;i++){
		e[fa[i]].push_back(i);
	}
	dfs(cnt,1);
	memset(l,0,sizeof(l));memset(r,0,sizeof(r));
	for(int i=1;i<=n;i++)e[i].clear();sz[1]=0;
}

整体代码也很简单

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e3+5;
inline int read(){
	char c=getchar();int t=0,f=1;
	while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
	while(isdigit(c)){t=(t<<3)+(t<<1)+(c^48);c=getchar();}
	return t*f;
}
int num,m,n,fa[maxn],pa[maxn],fb[maxn],pb[maxn];
int l[maxn],r[maxn];
long long val[2][maxn][maxn],dp[maxn],sz[maxn];
vector<int> e[maxn];
void dfs(int cnt,int u){
	if(u!=1)sz[u]=1;
	for(int i=0;i<e[u].size();i++){
		int v=e[u][i];
		dfs(cnt,v);sz[u]+=sz[v];
		l[u]=min(l[u],l[v]);
		r[u]=max(r[u],r[v]);
	}
	val[cnt][l[u]][r[u]]=max(val[cnt][l[u]][r[u]],sz[u]);
}
void solve(int cnt){
	n=read();
	for(int i=1;i<=n;i++)l[i]=n+1;
	for(int i=2;i<=n;i++)fa[i]=read();
	for(int i=1;i<=num;i++){
		pa[i]=read();
		l[pa[i]]=r[pa[i]]=i;
	}
	for(int i=2;i<=n;i++){
		e[fa[i]].push_back(i);
	}
	dfs(cnt,1);
	memset(l,0,sizeof(l));memset(r,0,sizeof(r));
	for(int i=1;i<=n;i++)e[i].clear();sz[1]=0;
}
int main(){
	num=read();
	solve(0);solve(1);
	for(int i=1;i<=num;i++){
		for(int j=i;j<=num;j++){
			dp[j]=max(dp[j],dp[i-1]+max(val[0][i][j],val[1][i][j]));
		}
	}
	printf("%lld\n",dp[num]);
	return 0;
}