【bzoj1468】Tree

原创 2016年06月01日 17:04:07

*题目描述:
给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K
*输入:
N(n<=40000) 接下来n-1行边描述管道,按照题目中写的输入 接下来是k
*输出:
一行,有多少对点之间的距离小于等于k
*样例输入:
7
1 6 13
6 3 9
3 5 7
4 1 3
2 4 20
4 7 2
10
*样例输出:
5
*题解:
点分治。就是把树上的路径统计问题分成过某个顶点和不过这个点的来做。不过这个点的就递归下去做,过这个点的就直接做。每次递归的时候找到这棵树的重心然后继续往下做,这样时间复杂度就有保证了,每一次找到重心后,这棵树的size至少缩小了一半,于是乎点分治的时间复杂度则降为了O(nlog2n)。不过这题统计过一个点的路径时需要排序(貌似可以用基排?反正我懒得手写),于是乎这题的复杂度就为O(nlog22n)的了(如果是基数排序的话就为O(nlog2n))。
*代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>

#ifdef WIN32
    #define LL "%I64d"
#else
    #define LL "%lld"
#endif

#ifdef CT
    #define debug(...) printf(__VA_ARGS__)
    #define setfile() 
#else
    #define debug(...)
    #define filename ""
    #define setfile() freopen(filename".in", "r", stdin); freopen(filename".out", "w", stdout);
#endif

#define R register
#define getc() (S == T && (T = (S = B) + fread(B, 1, 1 << 15, stdin), S == T) ? EOF : *S++)
#define dmax(_a, _b) ((_a) > (_b) ? (_a) : (_b))
#define dmin(_a, _b) ((_a) < (_b) ? (_a) : (_b))
#define cmax(_a, _b) (_a < (_b) ? _a = (_b) : 0)
#define cmin(_a, _b) (_a > (_b) ? _a = (_b) : 0)
char B[1 << 15], *S = B, *T = B;
inline int FastIn()
{
    R char ch; R int cnt = 0; R bool minus = 0;
    while (ch = getc(), (ch < '0' || ch > '9') && ch != '-') ;
    ch == '-' ? minus = 1 : cnt = ch - '0';
    while (ch = getc(), ch >= '0' && ch <= '9') cnt = cnt * 10 + ch - '0';
    return minus ? -cnt : cnt;
}
#define maxn 300010
#define maxm 600010
struct Edge
{
    Edge *next;
    int to, w;
}*last[maxn], e[maxm], *ecnt = e;
inline void link(R int a, R int b, R int v)
{
    *++ecnt = (Edge) {last[a], b, v}; last[a] = ecnt;
}
int n, k, size[maxn], son[maxn], root, sum, depcnt;
long long ans, deep[maxm], d[maxn];
bool vis[maxn];
void dfsroot(R int x, R int fa)
{
    size[x] = 1; son[x] = 0;
    for (R Edge *iter = last[x]; iter; iter = iter -> next)
    {
        R int pre = iter -> to;
        if (pre == fa || vis[pre]) continue;
        dfsroot(pre, x);
        size[x] += size[pre];
        cmax(son[x], size[pre]);
    }
    cmax(son[x], sum - size[x]);
    if (root == 0 || son[x] < son[root]) root = x;
}
void dfsdep(R int x, R int fa)
{
    deep[++depcnt] = d[x];
    for (R Edge *iter = last[x]; iter; iter = iter -> next)
    {
        R int pre = iter -> to;
        if (pre == fa || vis[pre]) continue;
        d[pre] = d[x] + iter -> w;
        dfsdep(pre, x);
    }
}
inline long long cal(R int x, R int val)
{
    d[x] = val; depcnt = 0;
    dfsdep(x, 0);
    std::sort(deep + 1, deep + depcnt + 1);
    R long long t = 0;
    R int l, r;
    for (l = 1, r = depcnt; l < r; )
    {
        if (deep[l] + deep[r] <= k) {t += r - l; ++l;}
        else --r;
    }
    return t;
}
void work(R int x)
{
    ans += cal(x, 0);
    vis[x] = 1;
    for (R Edge *iter = last[x]; iter; iter = iter -> next)
    {
        R int pre = iter -> to;
        if (vis[pre]) continue;
        ans -= cal(pre, iter -> w);
        sum = size[pre];
        root = 0;
        dfsroot(pre, 0);
        work(root);
    }
}
int main()
{
    n = FastIn(); sum = n;
    for (R int i = 1; i < n; ++i)
    {
        R int a = FastIn(), b = FastIn(), v = FastIn();
        link(a, b, v); link(b, a, v);
    }
    k = FastIn();
    son[root = 0] = 1 << 30;
    dfsroot(1, 0);
    work(root);
    printf("%lld\n", ans );
    return 0;
}

相关文章推荐

【BZOJ1468】Tree

TREE 点分治 男人八题

【BZOJ】【P1468】【Tree】【题解】【点分治】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1468 刚学点分治,简要的记录一下...

BZOJ 1468: Tree

Description 给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K Input N(n Output 一行,有多少对点之间的距离小于等于k ...

bzoj 1468: Tree

Description 给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K Input N(n Output 一行,有多少对点之间的距离小于等于k ...
  • lqybzx
  • lqybzx
  • 2015年04月02日 09:56
  • 300

【bzoj1468】【Tree】【点分治】

Description 给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K Input N(n Output 一行,有多少对点之间的距离小于等于k ...

【点分治】BZOJ 1468:Tree

BZOJ 1468:TreeDescription给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于KInputN 接下来n-1行边描述管道,按照题目中写的输入 接下来是kO...

POJ 1741|BZOJ 1468|Tree|树分治

求树上距离不超过K的点对数。 一条链有2种情况。 一是经过根;二是在子树中。 为了排除情况二,我们统计两端点不在同一子树中的。 可以先统计所有,再统计在同一子树中的。 于是可以对所有点距离排...

bzoj1468 tree 点分治

给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K 计算经过根的路径,既离根的距离之和小于K的点,然后递归处理不经过根的路径。这样会有一些重复,两点与根的距离之和小于K,但...
  • Loi_a
  • Loi_a
  • 2016年12月20日 19:16
  • 177

poj 1741 bzoj1468 Tree

Description Give a tree with n vertices,each edge has a length(positive integer less than 1001). Def...

【bzoj1592】【Usaco2008 Feb】【动态规划】CQYZ_Vijos_P1468 路面修整

【问题描述】  FJ打算好好修一下农场中某条凹凸不平的土路。按奶牛们的要求,修好后的路面高度应当单调上升或单调下降,也就是说,高度上升与高度下降的路段不能同时出现在修好的路中。   整条路被分成了 ...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:【bzoj1468】Tree
举报原因:
原因补充:

(最多只允许输入30个字)