【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

树的点分治
  • AaronGZK
  • AaronGZK
  • 2016年05月13日 00:24
  • 1197

【bzoj1468】【Tree】【点分治】

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

bzoj1468: Tree 点分治

今天考试,依旧打暴力咯。一看第一题绝壁是个虚树之类的东东。就是给你一棵树每次选几个点,给定几个点的控制范围,统计有多少点被控制。正解是离线+点分治+虚树。搞得我整个人都BB了。不过这场真是暴力出奇迹,...
  • IED98
  • IED98
  • 2015年03月25日 19:03
  • 616

BZOJ1468 - Tree

点分治第一题
  • VicJiao
  • VicJiao
  • 2018年01月18日 23:49
  • 26

bzoj1468: Tree

点分治
  • FSAHFGSADHSAKNDAS
  • FSAHFGSADHSAKNDAS
  • 2017年04月23日 10:59
  • 140

【BZOJ1468】Tree

TREE 点分治 男人八题
  • johann_oier
  • johann_oier
  • 2015年10月21日 13:28
  • 307

bzoj1468 Tree(点分治)

点分治,就是每次找到目前子树的重心,把它当做根,处理所有过重心的路径信息,然后递归处理子树。 什么是一棵树的重心呢?选出的一个点,使分出的子树中节点数最大的子树最小,就是重心。这样分出的子树点数不超...
  • Icefox_zhx
  • Icefox_zhx
  • 2018年01月16日 10:49
  • 60

[BZOJ1468]Tree 点分治

点分治模板。 易错点:分治时候传进函数里的size应为(sz[p->t]>sz[rt]?size-sz[rt]:sz[p->t]) 代码:#include #include #include us...
  • DOFYPXY
  • DOFYPXY
  • 2017年12月11日 10:47
  • 58

[bzoj1468]Tree(点分治)

传送门 点分治这个算法一开始学的时候真的觉得这就是玄学,因为按道理来说,这样递归应该是n2lognn^2logn的呀,为什么就是nlog2nnlog^2n呢?然后我就想到了树链剖分这个神奇的算法,想...
  • stone41123
  • stone41123
  • 2017年09月30日 12:25
  • 117

poj 1741 bzoj1468 Tree

Description Give a tree with n vertices,each edge has a length(positive integer less than 1001). Def...
  • lzr010506
  • lzr010506
  • 2016年02月28日 14:55
  • 210
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:【bzoj1468】Tree
举报原因:
原因补充:

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