关闭

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

1309人阅读 评论(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;
}


0
0
查看评论
发表评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场

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
  • 2183

[bzoj4543/3522]Hotel

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

[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
  • 184

BZOJ3522&4543: [POI2014]Hotel加强版

题解见http://blog.csdn.net/neither_nor/article/details/51278993 不知道这是什么狗题,时间复杂度貌似是O(N) #include #includ...
  • commonc
  • commonc
  • 2016-05-11 19:13
  • 1291

BZOJ4543: [POI2014]Hotel加强版 长链剖分

给一个树,问有多少三元组满足两两距离相等。n<=100000 长链剖分应用之一:o(n)统计以深度为下标的可合并子树信息 在当前节点,令f(i)表示相对深度为i的节点个数,g(i)表示在子树外离当...
  • Mima_Reincarnation
  • Mima_Reincarnation
  • 2017-04-11 17:42
  • 801

bzoj4543

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

BZOJ 3522 & 4543: [POI2014]Hotel

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

关于treap启发式合并的一点脑洞(以bzoj2809为例)

首先我直到bzoj2809正解应该是可并堆,之所以写treap启发式合并单纯只是因为这个脑洞… 首先我们有两个treap,分别是 A 和 B ,它们的节点数分别为 n 和 m (n<m)(n<m)。网...
  • scpointer
  • scpointer
  • 2017-01-30 22:28
  • 317

【bzoj3522】【poi2014】【hotel】【树形dp】

Description 有一个树形结构的宾馆,n个房间,n-1条无向边,每条边的长度相同,任意两个房间可以相互到达。吉丽要给他的三个妹子各开(一个)房(间)。三个妹子住的房间要互不相同(否则要打起来...
  • sunshinezff
  • sunshinezff
  • 2016-07-08 09:59
  • 208

线段树专题—POJ 3667 Hotel(区间合并模板)

题意:给一个n和m,表示n个房间,m次操作,操作类型有2种,一种把求连续未租出的房间数有d个的最小的最左边的房间号,另一个操作时把从x到x+d-1的房间号收回。 分析:这是一个区间合并的典型...
  • sin_XF
  • sin_XF
  • 2015-08-15 14:47
  • 707
    个人资料
    • 访问:337091次
    • 积分:12744
    • 等级:
    • 排名:第1262名
    • 原创:975篇
    • 转载:3篇
    • 译文:0篇
    • 评论:54条
    文章分类
    最新评论