bzoj-3522 Hotel

题意:
在一颗n个结点的树上给吉丽的三个妹子各开一个房间,使三个房间两两距离相等;
n<=5000,树上路径长度均为1;


题解:
首先因为树上两点间只有一条路径,所以这种路径下满足条件的三点只可能形成 以某个点为中心,三个点都到那个顶点距离相同的情况;
然后我们枚举每个点深搜,找出所有深度相同的点然后累乘C[x][3];
然后发现这么做是不对的!
因为我们将同一颗子树中深度相同的点也计入了答案中,它们之间的距离是不满足题中要求的;
那么换一种方式,每次统计根节点的某儿子的子树中深度一定的点,然后跑一个DP,状态为f[i][j][k]表示前i个儿子中深度和为j,选了k个的方案数,第一维可以滚动,累加k==3的情况就是答案啦;
注意一下开long long,时间复杂度不要写搓了就是O(n^2);


代码:


#include<stdio.h>
#include<string.h>
#include<algorithm>
#define N 5100
using namespace std;
typedef long long ll;
int next[N<<1],to[N<<1],head[N],ce;
int deep[N],hash[N],last;
ll f1[N],f2[N];
void add(int x,int y)
{
	to[++ce]=y;
	next[ce]=head[x];
	head[x]=ce;
}
void dfs(int x,int pre)
{
	deep[x]=deep[pre]+1;
	hash[deep[x]]++;
	last=max(last,deep[x]);
	for(int i=head[x];i;i=next[i])
	{
		if(to[i]!=pre)
		{
			dfs(to[i],x);
		}
	}
}
int main()
{
	int n,m,i,j,k,x,y;
	ll ans;
	scanf("%d",&n);
	for(i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	for(i=1,ans=0;i<=n;i++)
	{
		deep[i]=0;
		memset(f1,0,sizeof(f1));
		memset(f2,0,sizeof(f2));
		for(k=head[i];k;k=next[k])
		{
			memset(hash,0,sizeof(int)*(last+5));
			last=0;
			dfs(to[k],i);
			for(j=1;j<=last;j++)
			{
				ans+=hash[j]*f2[j];
				f2[j]+=f1[j]*hash[j];
				f1[j]+=hash[j];
			}	
		}
	}
	printf("%lld\n",ans);
	return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值