BZOJ3522: [Poi2014]Hotel Dfs

最新推荐文章于 2019-08-01 21:06:29 发布

Oakley_

最新推荐文章于 2019-08-01 21:06:29 发布

阅读量682

点赞数 1

文章标签： Dfs

本文链接：https://blog.csdn.net/Oakley_/article/details/52327962

版权

3522: [Poi2014]Hotel

Time Limit: 20 Sec Memory Limit: 128 MB

Submit: 502 Solved: 293

[Submit][Status][Discuss]

Description

有一个树形结构的宾馆，n个房间，n-1条无向边，每条边的长度相同，任意两个房间可以相互到达。吉丽要给他的三个妹子各开（一个）房（间）。三个妹子住的房间要互不相同（否则要打起来了），为了让吉丽满意，你需要让三个房间两两距离相同。

有多少种方案能让吉丽满意？

Input

第一行一个数n。
接下来n-1行，每行两个数x,y，表示x和y之间有一条边相连。

Output

让吉丽满意的方案数。

Sample Input

7
1 2
5 7
2 5
2 3
5 6
4 5

Sample Output

HINT

【样例解释】

{1,3,5},{2,4,6},{2,4,7},{2,6,7},{4,6,7}

【数据范围】

n≤5000

题目大意：给定一棵树，找到点组成的三元组(x,y,z)使dis(x,y)=dis(x,z)=dis(y,z);

题解：
对于这样的三元组，一定有一个点到这三个点的距离都相等，于是我们可以暴力搜索每一个点
以x点为树根：
三个点可能分布在不同的子树中，也可能分布在同一个子树中
但是对于分布在同一颗子树中的情况我们可以在后来的点中搜索到，所以不必考虑
我们直接搜索每一棵子树中距离x的距离相等的有多少个
f[i]表示与x距离为i的点有多少个，g[i]表示与x距离相等的点对有多少个

<span style="font-size:14px;"><span style="font-family:SimSun;">#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const ll M=15050;
ll n,dep[M],f[M],g[M],ans;
ll cnt,to[M],nxt[M],lj[M];
void add(ll f,ll t)
{
	cnt++;
	to[cnt]=t;
	nxt[cnt]=lj[f];
	lj[f]=cnt;
}
void Dfs(ll x,ll from,ll dpt)
{
	dep[dpt]++;
	for(ll i=lj[x];i;i=nxt[i])
	if(to[i]!=from) Dfs(to[i],x,dpt+1);
}
ll x,y;
int main()
{
	scanf("%lld",&n);
	for(ll i=1;i<n;i++)
	{
		scanf("%lld%lld",&x,&y);
		add(x,y);
		add(y,x);
	}
	for(x=1;x<=n;x++)
	{
		memset(f,0,sizeof(f));
		memset(g,0,sizeof(g));
		for(ll i=lj[x];i;i=nxt[i])
		{
			memset(dep,0,sizeof(dep));
			Dfs(to[i],x,1);
			for(ll j=1;j<=n;j++)
			{
				ans+=(ll)g[j]*dep[j];
				g[j]+=f[j]*dep[j];
				f[j]+=dep[j];
			}
		}
	}
	printf("%lld",ans);
}
</span></span>