[树形DP 启发式合并 神题] BZOJ 4543 [POI2014]Hotel加强版 & BZOJ 3522 [Poi2014]Hotel

1501人阅读 评论(1) 收藏 举报
分类:

%%%神犇的题解

转移不难想 关键是空间和时间

空间用轻重链的思想和指针转移 

时间么 指针转移&启发式合并

复杂度的证明比较巧妙


任意设一点作为根。令 f(a, d) 表示在以 a 点为根的子树中,与 a 距离为 d 的节点数;g(a, d) 表示在以

a 为根的子树中选择两个节点,满足剩下的一个节点 s 需在 a 子树外选择且与 a 的距离必须为 d 的方案数。
则方案数容易统计,且不难写出 DP 转移式。
注意到 f(a, d), g(a, d) 满足 d ≤ size(a),则转移式可以用启发式合并优化。再注意到若 a 只有一个儿子
b ,则:
f(a, x) = f(b, x ? 1)
g(a, x) = g(b, x + 1)
即 f(a) = f(b) ? 1, g(a) = g(b) + 1,这一步可以利用指针在 O(1) 内实现。则可以在 O(1) 时间内将某个
点 size 最大的儿子的 DP 数组转移到该点上。利用类似树链剖分的方式可以预处理出所需要的数组空间。
这个看似是nlog的
但实际上如果一个点v 在某个祖先u 对复杂度有+1的效果,说明v 不在u往下的最长链上,并且v 在 u 的儿子u’(也是 v 的祖先)的最长链上。则到u 的父亲时,v 不在 u的最长链上,则对答案没有贡献。所以贡献的总和是 O(n)的。


#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define V G[p].v
using namespace std;
typedef long long ll;

inline char nc()
{
	static char buf[100000],*p1=buf,*p2=buf;
	if (p1==p2) { p2=(p1=buf)+fread(buf,1,100000,stdin); if (p1==p2) return EOF; }
	return *p1++;
}

inline void read(int &x){
	char c=nc(),b=1;
	for (;!(c>='0' && c<='9');c=nc()) if (c=='-') b=-1;
	for (x=0;c>='0' && c<='9';x=x*10+c-'0',c=nc()); x*=b;
}

const int N=100005;

struct edge{
	int u,v,next;
};

edge G[N<<1];
int head[N],inum;

inline void add(int u,int v,int p){
	G[p].u=u; G[p].v=v; G[p].next=head[u]; head[u]=p;
}

int n;
int depth[N],bot[N],son[N];
ll Mem[N*10],*pnt=Mem;
ll *f[N],*g[N];
ll ans;

inline void dfs(int u,int fa){
	bot[u]=u; depth[u]=depth[fa]+1;
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa){
			dfs(V,u);
			if (depth[bot[V]]>depth[bot[u]])
				bot[u]=bot[V],son[u]=V;
		}
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa && (u==1 || V!=son[u])){
			int v=bot[V];
			f[v]=(pnt+=depth[v]-depth[u]+1); 
			g[v]=(++pnt);
			pnt+=((depth[v]-depth[u])<<1)+10;  
		}
}

inline void dp(int u,int fa){
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa){
			dp(V,u);
			if (V==son[u])
				f[u]=f[V]-1,g[u]=g[V]+1;
		}
	ans+=g[u][0]; f[u][0]=1;
	for (int p=head[u];p;p=G[p].next)
		if (V!=fa && V!=son[u]){
			for(int j=0;j<=depth[bot[V]]-depth[u];j++)
				ans+=f[u][j-1]*g[V][j]+g[u][j+1]*f[V][j];
			for(int j=0;j<=depth[bot[V]]-depth[u];j++){  
				g[u][j-1]+=g[V][j];  
				g[u][j+1]+=f[u][j+1]*f[V][j];  
				f[u][j+1]+=f[V][j];
			}
		}
}

int main()
{
	int iu,iv;
	freopen("t.in","r",stdin);
	freopen("t.out","w",stdout);
	read(n);
	for (int i=1;i<n;i++)
		read(iu),read(iv),add(iu,iv,++inum),add(iv,iu,++inum);
	dfs(1,0); dp(1,0);
	printf("%lld\n",ans);
	return 0;
}


查看评论

BZOJ4543/BZOJ3522 [POI2014]Hotel加强版

这题真TM神…… 和commonc一起orz了半天题解-_- 我们考虑一下dp,f[i][j]表示以i为根的子树里与i距离为j的点的个数,g[i][j]表示以i为根的子树里有这么多个点对在底下分叉了,...
  • neither_nor
  • neither_nor
  • 2016-04-29 07:33:05
  • 2627

[bzoj4543/3522]Hotel

题目大意一颗n个节点的树。 找三个不同编号的节点,使它们两两间距离相同(一条边距离视作1),求方案数。 在3522的版本中,n...
  • WerKeyTom_FTD
  • WerKeyTom_FTD
  • 2017-02-28 12:26:03
  • 672

[树形DP 启发式合并 神题] BZOJ 4543 [POI2014]Hotel加强版 & BZOJ 3522 [Poi2014]Hotel

%%%神犇的题解 转移不难想 关键是空间和时间 空间用轻重链的思想和指针转移  时间么 指针转移&启发式合并 复杂度的证明比较巧妙 “任意设一点作为根。令 f(a, d) 表示在以 ...
  • u014609452
  • u014609452
  • 2016-05-18 21:31:46
  • 1501

bzoj4543

题意: 给出一棵树,每条边长度为1。问找三个点,使他们两两距离相等有多少方案。 n 题解: 膜了题解。。太神了转述一下 显然对于每个方案(x,y,z)有且仅有一个点u使得: 1、x,y,z在...
  • u010600261
  • u010600261
  • 2016-12-12 13:20:53
  • 600

【XSY1536】【BZOJ3522】【BZOJ4543】【POI2014】Hotel 树形DP 长链剖分 启发式合并

题目描述​  给你一棵树,求有多少个组点满足x≠y,x≠z,y≠z,distx,y=distx,z=disty,zx\neq y,x\neq z,y\neq z,dist_{x,y}=dist_{x,...
  • ez_yww
  • ez_yww
  • 2017-08-12 12:47:15
  • 227

BZOJ 3522 & 4543: [POI2014]Hotel

指针优化树形DPBZOJ3522的n只有5000,可以随便用一个n^2的大暴力搞过去,就不讲了。BZOJ4543好丧啊,看到题解我当时就震惊了,题解点我n有十万,使人一看到就会放弃一些树形DP的思路,...
  • ziqian2000
  • ziqian2000
  • 2016-10-21 23:48:41
  • 229

【DP-树形DP】BZOJ3522/BZOJ4543 [POI2014]Hotel

这个dp是真的妙啊qwq 我永远喜欢薇尔谢特!
  • Dream_Lolita
  • Dream_Lolita
  • 2018-02-16 21:20:28
  • 56

BZOJ4675 点对游戏

根据期望的线性性(orz xuruifan,我们一定要叫线性性,不能叫可加性),期望幸运点对数等于每个幸运点对被选中的概率之和 由于每个点都是等价的,所以每个点对也是等价的,所以我们只需要计算1个点...
  • neither_nor
  • neither_nor
  • 2016-08-10 16:37:05
  • 397

[bzoj3522][bzoj4543][POI2014]HOTEL

Description给出一棵树,求这棵树中有多少种方法选择三个点使得这三个点之间的距离两两相等。 n...
  • alan_cty
  • alan_cty
  • 2017-09-11 22:22:38
  • 213

[bzoj4543] [POI2014]Hotel加强版

题目大意给定一棵树,n个节点,问有多少个三元组(x,y,z)(x < y < z),满足这三个点在树上距离两两相等。 n≤100000分析你可以想到一个n方的dp:设f[i][j]表示i为根的子树中...
  • WorldWide_D
  • WorldWide_D
  • 2017-09-13 21:37:57
  • 273
    个人资料
    持之以恒
    等级:
    访问量: 40万+
    积分: 1万+
    排名: 1233
    文章分类
    最新评论