标签：加强版 int 复杂度 100100 POI2014 HOT 距离 LCA rightarrow

I.II.[POI2014]HOT-Hotels 加强版

易于发现，三点间两两距离相等，当且仅当两个LCA较深的点到它们的LCA距离相等，且第三个点到该LCA的距离与前两个点相同。

于是我们设 \(f_{x,i}\) 表示 \(x\) 子树中有多少对点，它们到LCA距离相同，且上述距离比该LCA到 \(x\) 的距离长 \(i\)。考虑现在将子树 \(y\) 合并进答案。

则有 \(f_{x,i}\times g_{y,i+1}\rightarrow ans\)，\(g_{x,i}\times f_{y,i-1}\rightarrow ans\)，\(f_{x,i}\times f_{y,i-1}\rightarrow g_{x,i}\)，\(f_{y,i-1}\rightarrow f_{x,i}\)，\(g_{y,i+1}\rightarrow g_{x,i}\)。

但是这显然过不去 \(10^5\) 的数据。

但是，因为DP状态与深度有关，就可以考虑长链剖分优化。具体而言，考虑到仅有一个儿子时，直接有 \(f_{y,i-1}\rightarrow f_{x,i}\) ，\(g_{y,i+1}\rightarrow g_{x,i}\)，以及一个 \(g_{y,1}\rightarrow ans\)，可以直接通过指针之类 \(O(1)\) 转移。则如果我们考虑继承长儿子的答案的话，就会大大优化复杂度。具体优化了多少呢？总复杂度可以被证明是 \(O(n)\) 的。

指针使用时要极其小心，不能调用奇奇怪怪的东西，不能越界，一定要仔细算好范围。

时间复杂度 \(O(n)\)。

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,dep[100100],son[100100];
int F[500100],*f[100100],*ff=F;
ll G[500100],*g[100100],res,*gg=G;
vector<int>v[100100];
void dfs1(int x,int fa){
	for(auto y:v[x])if(y!=fa){dfs1(y,x);if(dep[son[x]]<dep[y])son[x]=y;}
	dep[x]=dep[son[x]]+1;
}
void init(int x){f[x]=ff,ff+=dep[x]+1,gg+=dep[x]+1,g[x]=gg,gg+=dep[x]+1;}
void dfs2(int x,int fa){
	if(!f[x])f[x]=ff,ff+=(dep[x]<<1);
	if(!g[x])gg+=dep[x],g[x]=gg,gg+=dep[x];
	if(son[x])f[son[x]]=f[x]+1,g[son[x]]=g[x]-1,dfs2(son[x],x),res+=g[x][0];
	f[x][0]=1;
	for(auto y:v[x])if(y!=fa&&y!=son[x]){
		dfs2(y,x);
		for(int i=1;i<=dep[y];i++)res+=g[y][i]*f[x][i-1];
		for(int i=0;i<=dep[y];i++)res+=g[x][i+1]*f[y][i];
		for(int i=0;i<=dep[y];i++)g[x][i+1]+=1ll*f[x][i+1]*f[y][i];
		for(int i=0;i<dep[y];i++)g[x][i]+=g[y][i+1];
		for(int i=0;i<=dep[y];i++)f[x][i+1]+=f[y][i];
	}
//	printf("%d:\n",x);for(int i=0;i<=dep[x];i++)printf("%d %d\n",f[x][i],g[x][i]);
}
int main(){
	scanf("%d",&n);
	for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),v[x].push_back(y),v[y].push_back(x);
	dep[0]=-1,dfs1(1,0);
//	for(int i=1;i<=n;i++)printf("(%d %d)",dep[i],son[i]);puts("");
	dfs2(1,0),printf("%lld\n",res);
	return 0;
}

标签：加强版,int,复杂度,100100,POI2014,HOT,距离,LCA,rightarrow
来源： https://www.cnblogs.com/Troverld/p/14636986.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。