skkyk:点分治

由题开始==

例题:求在一棵有权树上,是否存在一条路径满足权值和为K
解法:以每个点为根一次,看在他的子树间是否存在两段,其和为K;O(==)

和例题一样,对于树上问题,求某些要求的路径(数量或者存在性等),
往往可以先对一条经过根节点的路径操作,
后再删去这个根,对他的子树们同样的操作
显然是一个分治过程
原理就是,一条路径,要么是由一个点经过根节点,与其他子树内的节点形成;
要么就是只在这棵子树内形成路径
大概图示意思(红绿为两条上述路径)
skkyk:点分治_第1张图片
<

当我们的树比较平衡时,每个点被路径计算是$ logn $ 的,但是当树是一条链的时候,就退化成$ n^2 $ 了
为了避免这种情况,可以用树的重心代替,成为新的根。此时总复杂度为$ O (nlogn) 。$
原因就是,一棵树,怎么为根都还是一颗树,但是以重心为根的时候,这颗树是最好看的最平衡的
rt,将链按箭头提起来:
skkyk:点分治_第2张图片
skkyk:点分治_第3张图片
这看向去更像是一棵树

所以

总结一下
一棵树先确定他的重心,以重心为根,确定经过根节点的路径;再把根节点删掉,对于删掉他的子树们,按上述同样操作;
已证$ O(nlogn) $。
luogu模板题代码仅供参考,不解释。

#include
using namespace std;
const int N = 1e4+50;
const int K = 1e7+50;
int n,m;
struct node{int next,to,dis;}edge[N<<1];
int head[N],cnt;
inline void add(int from,int to,int dis) {
    edge[++cnt].to=to,edge[cnt].dis=dis,edge[cnt].next=head[from],head[from]=cnt;
}
int q[N],ans[N],maxp[N],size[N],visited[N],tmp[N],dis[N],judge[K];
int rt,sum,tot;
void getrt(int u,int f) {
    size[u]=1,maxp[u]=0;
    for(int i=head[u];i;i=edge[i].next) {
        int v=edge[i].to;if(v==f||visited[v]) continue;
        getrt(v,u);
        size[u]+=size[v];maxp[u]=max(maxp[u],size[v]);
    }
    maxp[u]=max(maxp[u],sum-maxp[u]);
    if(maxp[u] que;
void solve(int u) {
    for(int i=head[u];i;i=edge[i].next) {
        int v=edge[i].to;if(visited[v]) continue;
        dis[v]=edge[i].dis;
        tot=0;getdis(v,u);
        for(int j=1;j<=tot;j++) 
            for(int k=1;k<=m;k++) 
                if(q[k]>=tmp[j])
                    ans[k]|=judge[q[k]-tmp[j]];
        for(int j=1;j<=tot;j++) que.push(tmp[j]),judge[tmp[j]]=1;
    }
    while(!que.empty()) judge[que.front()]=0,que.pop();//数组过大,memset超时 
}
void divide(int u) {
    judge[0]=visited[u]=1;solve(u);
    for(int i=head[u];i;i=edge[i].next) {
        int v=edge[i].to;if(visited[v]) continue;
        maxp[rt=0]=sum=size[v];
        getrt(v,0),getrt(rt,0);
        divide(rt);
    }
}
int main() {
    cin>>n>>m;
    for(int i=1;i

你可能感兴趣的:(skkyk:点分治)