树的路径统计
统计有多少条长度不大于k的路径
树的点分治:将路径分为经过重心和不经过重心的路径。
解法一:
1.统计经过重心的满足条件的路径条数。
1)统计经过重心,且路径可在重心周围有重合的满足条件的路径条数—O(n)
2)统计经过重心的子节点,且路径可在重心子节点周围有重合的满足路径条数O(n)
3)经过重心满足条件的路径条数=1)-2)中的路径条数(巧妙之处!)
2.不经过重心,将重心除去,分为x个子树,分别寻找每个子树的重心,重复上述步骤即可。由于重心的定义可知,递归层数不会超过O(logn) ,因此总体复杂度为O(n*logn)
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#define maxn 20010
#define lowbit(x) x&(-x)
#define mem(a,b) memset(a,b,sizeof(a))
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
int n;
int k;
struct node{
int to,next,w;
}edge[maxn*2];
int head[maxn],tot;
int sum,f[maxn],vis[maxn],son[maxn],core,deep[maxn],d[maxn];//getcore
int ans;
void init(){
mem(head,-1);
mem(vis,0);
tot=0;
ans=0;
}
void add(int a,int b,int c){
edge[tot].to=b,edge[tot].next=head[a],edge[tot].w=c;
head[a]=tot++;
}
void getcore(int u,int fa){
son[u]=1,f[u]=0;//每次更新前必须初始化!!!---dfs可以在搜索子树前在内部对点初始化
for(int i=head[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(v==fa||vis[v]) continue;//除去已找到的重心后求子树重心
getcore(v,u);
son[u]+=son[v],f[u]=max(f[u],son[v]);
}
f[u]=max(f[u],sum-son[u]);
if(f[u]<f[core]) core=u;//core记录该树重心
}
void getdeep(int u,int