POJ 1741 树的分治

最新推荐文章于 2020-12-18 23:33:16 发布

sdj222555

最新推荐文章于 2020-12-18 23:33:16 发布

阅读量9.9k

点赞数 10

分类专栏： POJ 数据结构文章标签：算法 n2

本文链接：https://blog.csdn.net/sdj222555/article/details/7893862

版权

该博客介绍了如何使用分治算法解决POJ 1741题，即求解树上距离小于等于K的点对数量。由于n2算法效率低下，博主提出通过寻找树的重心进行分治。重心是删除后子树节点数最少的节点。通过两次DFS确定重心，然后利用快速排序和相向搜索在O(n log n)时间内找出满足条件的点对，并排除同一子树内的无效点对。整体时间复杂度为O(n log n log n)。

摘要由CSDN通过智能技术生成

题意就是求树上距离小于等于K的点对有多少个

n2的算法肯定不行，因为1W个点

这就需要分治。可以看09年漆子超的论文

本题用到的是关于点的分治。

一个重要的问题是，为了防止退化，所以每次都要找到树的重心然后分治下去，所谓重心，就是删掉此结点后，剩下的结点最多的树结点个数最小。

每次分治，我们首先算出重心，为了计算重心，需要进行两次dfs，第一次把以每个结点为根的子树大小求出来，第二次是从这些结点中找重心

找到重心后，需要统计所有结点到重心的距离，看其中有多少对小于等于K，这里采用的方法就是把所有的距离存在一个数组里，进行快速排序，这是nlogn的，然后用一个经典的相向搜索O(n)时间内解决。但是这些求出来满足小于等于K的里面只有那些路径经过重心的点对才是有效的，也就是说在同一颗子树上的肯定不算数的，所以对每颗子树，把子树内部的满足条件的点对减去。

最后的复杂度是n logn logn 其中每次快排是nlogn 而递归的深度为logn

#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
#include <cmath>
#include <queue>
#include <map>
#include <set>
#define eps 1e-5
#define MAXN 11111
#define MAXM 55555
#define INF 1000000000
using namespace std;
struct EDGE
{